SAm2

models/__init__.py

@@ -20,7 +20,6 @@
 import numpy as np
 import yaml
 import torch  # For memory management and CUDA operations
-import torch  # For memory management and CUDA operations
 
 # --------------------------------------------------------------------------------------
 # Logging
@@ -39,7 +38,7 @@
 # --------------------------------------------------------------------------------------
 # Path setup for third_party repos
 # --------------------------------------------------------------------------------------
-ROOT = Path(__file__).resolve().parent
+ROOT = Path(__file__).resolve().parent.parent  # Go up from models/ to project root
 TP_SAM2 = Path(os.environ.get("THIRD_PARTY_SAM2_DIR", ROOT / "third_party" / "sam2")).resolve()
 TP_MATANY = Path(os.environ.get("THIRD_PARTY_MATANY_DIR", ROOT / "third_party" / "matanyone")).resolve()
 
@@ -66,7 +65,6 @@ def _probe_ffmpeg() -> bool:
 
 def _has_cuda() -> bool:
     try:
-        import torch  # type: ignore
         return torch.cuda.is_available()
     except Exception:
         return False