matrice-analytics 0.1.41__py3-none-any.whl → 0.1.42__py3-none-any.whl

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
 

{matrice_analytics-0.1.41.dist-info → matrice_analytics-0.1.42.dist-info}/METADATA

@@ -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.dist-info → matrice_analytics-0.1.42.dist-info}/RECORD

@@ -166,7 +166,7 @@ matrice_analytics/post_processing/usecases/weld_defect_detection.py,sha256=b0dAJ
 matrice_analytics/post_processing/usecases/wildlife_monitoring.py,sha256=TMVHJ5GLezmqG7DywmqbLggqNXgpsb63MD7IR6kvDkk,43446
 matrice_analytics/post_processing/usecases/windmill_maintenance.py,sha256=G1eqo3Z-HYmGJ6oeZYrpZwhpvqQ9Lc_T-6S7BLBXHeA,40498
 matrice_analytics/post_processing/usecases/wound_segmentation.py,sha256=ehNX6VuWMB3xAnCySO3ra3Tf_5FUNg5LCSdq_91h374,38342
-matrice_analytics/post_processing/usecases/color/clip.py,sha256=qtG1YUngpt-5Emx4vdX1Hv_X9U6Sa4BMCNRDYpuM_4g,24828
+matrice_analytics/post_processing/usecases/color/clip.py,sha256=3pA_Oaw3hnMcuuyZLG5-1w1Kt04H04AHGy2d8463mzg,25741
 matrice_analytics/post_processing/usecases/color/color_map_utils.py,sha256=SP-AEVcjLmL8rxblu-ixqUJC2fqlcr7ab4hWo4Fcr_k,2677
 matrice_analytics/post_processing/usecases/color/color_mapper.py,sha256=nKPc28mJLlrl2HPua5EUUMweRRSI6WrrUBkeitTm7ms,17459
 matrice_analytics/post_processing/usecases/color/clip_processor/merges.txt,sha256=n9aR98gDkhDg_O0VhlRmxlgg0JtjmIsBdL_iXeKZBRo,524619
@@ -188,8 +188,8 @@ matrice_analytics/post_processing/utils/format_utils.py,sha256=UTF7A5h9j0_S12xH9
 matrice_analytics/post_processing/utils/geometry_utils.py,sha256=BWfdM6RsdJTTLR1GqkWfdwpjMEjTCJyuBxA4zVGKdfk,9623
 matrice_analytics/post_processing/utils/smoothing_utils.py,sha256=78U-yucAcjUiZ0NIAc9NOUSIT0PWP1cqyIPA_Fdrjp0,14699
 matrice_analytics/post_processing/utils/tracking_utils.py,sha256=rWxuotnJ3VLMHIBOud2KLcu4yZfDp7hVPWUtNAq_2xw,8288
-matrice_analytics-0.1.41.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
-matrice_analytics-0.1.41.dist-info/METADATA,sha256=aGpFsA4ZZ5rNPFR2iqDy8JlA-gAhpAtjniELau5yH1I,14378
-matrice_analytics-0.1.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matrice_analytics-0.1.41.dist-info/top_level.txt,sha256=STAPEU-e-rWTerXaspdi76T_eVRSrEfFpURSP7_Dt8E,18
-matrice_analytics-0.1.41.dist-info/RECORD,,
+matrice_analytics-0.1.42.dist-info/licenses/LICENSE.txt,sha256=_uQUZpgO0mRYL5-fPoEvLSbNnLPv6OmbeEDCHXhK6Qc,1066
+matrice_analytics-0.1.42.dist-info/METADATA,sha256=OLJohdXLb8RHXTcSxR5eUb1fcj85xYhpceBOMFEXnj4,14378
+matrice_analytics-0.1.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matrice_analytics-0.1.42.dist-info/top_level.txt,sha256=STAPEU-e-rWTerXaspdi76T_eVRSrEfFpURSP7_Dt8E,18
+matrice_analytics-0.1.42.dist-info/RECORD,,