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

View File

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

View File

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