diff --git a/machine-learning/app/models/constants.py b/machine-learning/app/models/constants.py index ce513a2f71..18965d2b1d 100644 --- a/machine-learning/app/models/constants.py +++ b/machine-learning/app/models/constants.py @@ -51,10 +51,10 @@ _INSIGHTFACE_MODELS = { } -SUPPORTED_PROVIDERS = {"CUDAExecutionProvider", "OpenVINOExecutionProvider", "CPUExecutionProvider"} +SUPPORTED_PROVIDERS = ["CUDAExecutionProvider", "OpenVINOExecutionProvider", "CPUExecutionProvider"] -STATIC_INPUT_PROVIDERS = {"OpenVINOExecutionProvider"} +STATIC_INPUT_PROVIDERS = ["OpenVINOExecutionProvider"] def is_openclip(model_name: str) -> bool: