ashipov/immich
1
0
Fork 1
forked from dark_thunder/immich
Code Pull Requests 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 073965c47e7e31f1255fd461cd34ee19917d78bb.

* Revert "cleanup"

This reverts commit bb56bd3297303332b25bdc3b112cfc278ee593ba.

* 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: