Merge pull request #1771 from roboflow/fix/depth-anythig-v2-fixes

Fix/depth anythig v2 fixes

Author: PawelPeczek-Roboflow

inference_experimental/inference_exp/models/auto_loaders/models_registry.py

@@ -14,6 +14,7 @@
 EMBEDDING_TASK = "embedding"
 CLASSIFICATION_TASK = "classification"
 MULTI_LABEL_CLASSIFICATION_TASK = "multi-label-classification"
+DEPTH_ESTIMATION_TASK = "depth-estimation"
 
 
 @dataclass(frozen=True)
@@ -353,6 +354,10 @@ class RegistryEntry:
         module_name="inference_exp.models.yolact.yolact_instance_segmentation_trt",
         class_name="YOLOACTForInstanceSegmentationTRT",
     ),
+    ("depth-anything-v2", DEPTH_ESTIMATION_TASK, BackendType.HF): LazyClass(
+        module_name="inference_exp.models.depth_anything_v2.depth_anything_v2_hf",
+        class_name="DepthAnythingV2HF"
+    )
 }
 
 

inference_experimental/inference_exp/models/depth_anything_v2/depth_anything_v2_hf.py

@@ -20,13 +20,17 @@ def from_pretrained(
         cls,
         model_name_or_path: str,
         device: torch.device = DEFAULT_DEVICE,
+        local_files_only: bool = True,
         **kwargs,
     ) -> "DepthAnythingV2HF":
-        model = AutoModelForDepthEstimation.from_pretrained(model_name_or_path).to(
-            device
-        )
+        model = AutoModelForDepthEstimation.from_pretrained(
+            model_name_or_path,
+            local_files_only=local_files_only,
+        ).to(device)
         processor = AutoImageProcessor.from_pretrained(
-            model_name_or_path, use_fast=True
+            model_name_or_path,
+            local_files_only=local_files_only,
+            use_fast=True
         )
         return cls(model=model, processor=processor, device=device)
 

inference_experimental/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "inference-exp"
-version = "0.16.4"
+version = "0.17.0rc2"
 description = "Experimental vresion of inference package which is supposed to evolve into inference 1.0"
 readme = "README.md"
 requires-python = ">=3.9,<3.13"