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

set log level

Browse Source
This commit is contained in:
mertalev 2025-03-14 23:37:39 -04:00
parent bc9bec8673
commit 76eb285204
No known key found for this signature in database
GPG Key ID: 8AC1FB7DEC668BCC
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
machine-learning/app/sessions/rknn/rknnpool.py
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")