From 9e689e835e5abd316e7133be0190feaab81f7700 Mon Sep 17 00:00:00 2001 From: mertalev <101130780+mertalev@users.noreply.github.com> Date: Sat, 15 Mar 2025 00:07:00 -0400 Subject: [PATCH] add rk3568 --- machine-learning/app/models/constants.py | 2 +- .../export/immich_model_exporter/exporters/constants.py | 2 +- machine-learning/export/immich_model_exporter/run.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/machine-learning/app/models/constants.py b/machine-learning/app/models/constants.py index b6c4daba23..d04ed4d543 100644 --- a/machine-learning/app/models/constants.py +++ b/machine-learning/app/models/constants.py @@ -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"] diff --git a/machine-learning/export/immich_model_exporter/exporters/constants.py b/machine-learning/export/immich_model_exporter/exporters/constants.py index ca72872858..4019119037 100644 --- a/machine-learning/export/immich_model_exporter/exporters/constants.py +++ b/machine-learning/export/immich_model_exporter/exporters/constants.py @@ -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 diff --git a/machine-learning/export/immich_model_exporter/run.py b/machine-learning/export/immich_model_exporter/run.py index f916b1d809..3554abb529 100644 --- a/machine-learning/export/immich_model_exporter/run.py +++ b/machine-learning/export/immich_model_exporter/run.py @@ -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}")