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}")