public/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-11-18 01:12:47 +09:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

fix(ml): openvino not working with dynamic axes (#6871)

Browse Source 
* convert to static

* add comment about gross code

* formatting

* fixed test

* fix typing

* cleanup

* formatting

* Revert "formatting"

This reverts commit 073965c47e.

* Revert "cleanup"

This reverts commit bb56bd3297.

* formatting
This commit is contained in:
Mert
2024-02-03 00:35:26 -05:00
committed by GitHub
parent b768eef44d
commit 79d3342c3d
2 changed files with 51 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -51,11 +51,10 @@ _INSIGHTFACE_MODELS = {
}
SUPPORTED_PROVIDERS = [
"CUDAExecutionProvider",
"OpenVINOExecutionProvider",
"CPUExecutionProvider",
]
SUPPORTED_PROVIDERS = ["CUDAExecutionProvider", "OpenVINOExecutionProvider", "CPUExecutionProvider"]
STATIC_INPUT_PROVIDERS = ["OpenVINOExecutionProvider"]
def is_openclip(model_name: str) -> bool: