matrice-analytics 0.1.41__tar.gz → 0.1.42__tar.gz

{matrice_analytics-0.1.41 → matrice_analytics-0.1.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_analytics
-Version: 0.1.41
+Version: 0.1.42
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_analytics-0.1.41 → matrice_analytics-0.1.42}/matrice_analytics.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matrice_analytics
-Version: 0.1.41
+Version: 0.1.42
 Summary: Common server utilities for Matrice.ai services
 Author-email: "Matrice.ai" <dipendra@matrice.ai>
 License-Expression: MIT

{matrice_analytics-0.1.41 → matrice_analytics-0.1.42}/src/matrice_analytics/post_processing/usecases/color/clip.py

@@ -247,15 +247,26 @@ def load_model_from_checkpoint(checkpoint_url: str, providers: Optional[List] =
         # Resolve providers
         available = ort.get_available_providers()
         print("Available providers:", available)
-        use_providers = ["CUDAExecutionProvider"] #providers or
 
-        # Validate providers and enforce CUDA when requested
-        if any(
-            (isinstance(p, tuple) and p[0] == "CUDAExecutionProvider") or p == "CUDAExecutionProvider"
-            for p in use_providers
-        ):
-            if "CUDAExecutionProvider" not in available:
-                raise RuntimeError("CUDAExecutionProvider not available in this environment")
+        # Use provided providers or default to CUDA if available, else use first available
+        if providers is None:
+            if "CUDAExecutionProvider" in available:
+                use_providers = ["CUDAExecutionProvider"]
+            elif available:
+                use_providers = [available[0]]
+                print(f"CUDA not available, using: {use_providers[0]}")
+            else:
+                use_providers = ["CPUExecutionProvider"]
+                print("No providers detected, using CPUExecutionProvider")
+        else:
+            use_providers = providers
+
+        # Warn if requested provider is not available
+        for provider in use_providers:
+            provider_name = provider[0] if isinstance(provider, tuple) else provider
+            if provider_name not in available:
+                print(f"Warning: Requested provider '{provider_name}' not in available providers: {available}")
+                print(f"Will attempt to use it anyway, may fall back to available providers")
 
         # Load ONNX model from bytes with enforced providers
         model = ort.InferenceSession(
@@ -330,8 +341,14 @@ class ClipProcessor:
             if "CUDAExecutionProvider" in available:
                 self.providers = ["CUDAExecutionProvider"]
             else:
-                # Enforce GPU-only per requirement; raise if not available
-                print("CUDAExecutionProvider not available; ensure CUDA-enabled onnxruntime-gpu is installed and GPU is visible")
+                # CUDA not available, fall back to CPU or other available providers
+                print("CUDAExecutionProvider not available; falling back to available providers")
+                if available:
+                    self.providers = [available[0]]  # Use first available provider
+                    print(f"Using provider: {self.providers[0]}")
+                else:
+                    self.providers = ["CPUExecutionProvider"]  # Ultimate fallback
+                    print("No providers detected, using CPUExecutionProvider")
         else:
             self.providers = providers