Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-07-09 03:04:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

ruff

Browse Source
This commit is contained in:
yoni13 2025-01-11 16:28:26 +08:00
parent d5e453a773
commit 23d0ea0e7b
3 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
machine-learning/app/models/base.py
View File

@ -8,11 +8,10 @@ from typing import Any, ClassVar
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
import ann.ann import ann.ann
import rknn.rknnpool
from app.sessions.ort import OrtSession from app.sessions.ort import OrtSession
from app.sessions.rknn import RknnSession from app.sessions.rknn import RknnSession
import rknn.rknnpool
from ..config import clean_name, log, settings from ..config import clean_name, log, settings
from ..schemas import ModelFormat, ModelIdentity, ModelSession, ModelTask, ModelType from ..schemas import ModelFormat, ModelIdentity, ModelSession, ModelTask, ModelType
from ..sessions.ann import AnnSession from ..sessions.ann import AnnSession

2
machine-learning/app/sessions/rknn.py
View File

@ -6,9 +6,9 @@ from typing import Any
import numpy as np import numpy as np
import onnxruntime as ort import onnxruntime as ort
from numpy.typing import NDArray from numpy.typing import NDArray
from rknn.rknnpool import rknnPoolExecutor
from app.schemas import SessionNode from app.schemas import SessionNode
from rknn.rknnpool import rknnPoolExecutor
from ..config import log from ..config import log

2
machine-learning/rknn/rknnpool.py
View File

@ -1,8 +1,8 @@
# This code is from leafqycc/rknn-multi-threaded # This code is from leafqycc/rknn-multi-threaded
# Following Apache License 2.0 # Following Apache License 2.0
from concurrent.futures import ThreadPoolExecutor
from queue import Queue from queue import Queue
from concurrent.futures import ThreadPoolExecutor, as_completed
supported_socs = ["rk3566", "rk3568", "rk3588"] supported_socs = ["rk3566", "rk3568", "rk3588"]