new: Provide userwarning when specifying providers and cuda (#425)

* new: Provide userwarning when specifying providers and cuda * Updated warning message
qdrant · Dec 24, 2024 · cec8d54 · cec8d54
1 parent 55b985c
commit cec8d54
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/fastembed/common/onnx_model.py b/fastembed/common/onnx_model.py
@@ -52,6 +52,13 @@ def _load_onnx_model(
         model_path = model_dir / model_file
         # List of Execution Providers: https://onnxruntime.ai/docs/execution-providers
 
+        if cuda and providers is not None:
+            warnings.warn(
+                f"`cuda` and `providers` are mutually exclusive parameters, cuda: {cuda}, providers: {providers}",
+                category=UserWarning,
+                stacklevel=6,
+            )
+
         if providers is not None:
             onnx_providers = list(providers)
         elif cuda: