SAM FIX2

models/__init__.py

@@ -255,7 +255,15 @@ def _resolve_sam2_cfg(cfg_str: str) -> str:
             return str(candidate)
     if cfg_path.exists():
         return str(cfg_path)
-    for name in ["configs/sam2/sam2_hiera_l.yaml", "configs/sam2/sam2_hiera_b.yaml", "configs/sam2/sam2_hiera_s.yaml"]:
+    # Updated SAM2 config search paths to match actual repository structure
+    for name in [
+        "sam2/configs/sam2.1/sam2.1_hiera_l.yaml",  # SAM 2.1 (preferred)
+        "sam2/configs/sam2/sam2_hiera_l.yaml",      # SAM 2.0
+        "sam2/configs/sam2.1/sam2.1_hiera_b.yaml", 
+        "sam2/configs/sam2/sam2_hiera_b.yaml",
+        "sam2/configs/sam2.1/sam2.1_hiera_s.yaml",
+        "sam2/configs/sam2/sam2_hiera_s.yaml"
+    ]:
         p = TP_SAM2 / name
         if p.exists():
             return str(p)
@@ -299,7 +307,7 @@ def load_sam2() -> Tuple[Optional[object], bool, Dict[str, Any]]:
         return None, False, meta
 
     device = _pick_device("SAM2_DEVICE")
-    cfg_env = os.environ.get("SAM2_MODEL_CFG", "configs/sam2/sam2_hiera_l.yaml")
+    cfg_env = os.environ.get("SAM2_MODEL_CFG", "sam2/configs/sam2/sam2_hiera_l.yaml")
     cfg = _resolve_sam2_cfg(cfg_env)
     ckpt = os.environ.get("SAM2_CHECKPOINT", "")