cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

set log level

Browse Source
This commit is contained in:
mertalev
2025-03-14 23:37:39 -04:00
parent bc9bec8673
commit 76eb285204
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/app/sessions/rknn/rknnpool.py
+2
View File
@@ -2,6 +2,7 @@
# Following Apache License 2.0 # Following Apache License 2.0
from concurrent.futures import Future, ThreadPoolExecutor from concurrent.futures import Future, ThreadPoolExecutor
import logging
from pathlib import Path from pathlib import Path
from queue import Queue from queue import Queue
from typing import Callable from typing import Callable
@@ -41,6 +42,7 @@ def init_rknn(model_path: str) -> "RKNNLite":
if not is_available: if not is_available:
raise RuntimeError("rknn is not available!") raise RuntimeError("rknn is not available!")
rknn_lite = RKNNLite() rknn_lite = RKNNLite()
rknn_lite.rknn_log.logger.setLevel(logging.ERROR)
ret = rknn_lite.load_rknn(model_path) ret = rknn_lite.load_rknn(model_path)
if ret != 0: if ret != 0:
raise RuntimeError("Failed to load RKNN model") raise RuntimeError("Failed to load RKNN model")