Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-05-24 01:12:58 -04:00
Code Issues Packages Projects Releases Wiki Activity

add rk3568

Browse Source
This commit is contained in:
mertalev 2025-03-15 00:07:00 -04:00
parent 76eb285204
commit 9e689e835e
No known key found for this signature in database
GPG Key ID: F7C271C07CF04AAE
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
machine-learning/app/models/constants.py
View File

@ -77,7 +77,7 @@ _INSIGHTFACE_MODELS = {
SUPPORTED_PROVIDERS = ["CUDAExecutionProvider", "OpenVINOExecutionProvider", "CPUExecutionProvider"]
RKNN_SUPPORTED_SOCS = ["rk3566", "rk3576", "rk3588"]
RKNN_SUPPORTED_SOCS = ["rk3566", "rk3568", "rk3576", "rk3588"]
RKNN_COREMASK_SUPPORTED_SOCS = ["rk3576", "rk3588"]

2
machine-learning/export/immich_model_exporter/exporters/constants.py
View File

@ -22,7 +22,7 @@ SOURCE_TO_METADATA = {
),
}
RKNN_SOCS = ["rk3566", "rk3576", "rk3588"]
RKNN_SOCS = ["rk3566", "rk3568", "rk3576", "rk3588"]
# glob to delete old UUID blobs when reuploading models

2
machine-learning/export/immich_model_exporter/run.py
View File

@ -77,7 +77,7 @@ def export_models(models: list[str], source: ModelSource) -> None:
for model in models:
try:
print(f"Exporting model {model}")
subprocess.check_call(["python", "export.py", model, source])
subprocess.check_call(["python", "-m", "immich_model_exporter.export", model, source])
except Exception as e:
print(f"Failed to export model {model}: {e}")