debug

examples/blobctrl/blobctrl_app.py

@@ -64,16 +64,14 @@ if not os.path.exists(Dino_path):
 print(f"Dino checkpoints downloaded to {Dino_path}")
 
 # download SAM model
-SAM_path = "examples/blobctrl/models/sam/sam_vit_h_4b8939.pth"
+SAM_path = "examples/blobctrl/models/sam"
 if not os.path.exists(SAM_path):
-    os.makedirs(os.path.dirname(SAM_path), exist_ok=True)
-    import urllib.request
-    print(f"Downloading SAM model...")
-    urllib.request.urlretrieve(
-        "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
-        SAM_path
+    SAM_path = snapshot_download(
+        repo_id="kunkaran/sam_vit_h_4b8939.pth",
+        local_dir=SAM_path,
+        token=os.getenv("HF_TOKEN"),
     )
-    print(f"SAM model downloaded to {SAM_path}")
+print(f"SAM checkpoints downloaded to {SAM_path}")
 
 
 ## load models and pipeline
@@ -81,7 +79,7 @@ blobnet_path = "./examples/blobctrl/models/blobnet"
 unet_lora_path = "./examples/blobctrl/models/unet_lora"
 stabel_diffusion_model_path = "./examples/blobctrl/models/stable-diffusion-v1-5"
 dinov2_path = "./examples/blobctrl/models/dinov2-large"
-sam_path = "./examples/blobctrl/models/sam/sam_vit_h_4b8939.pth"
+sam_path = "./examples/blobctrl/models/sam/sam_vt_h.pth"
 
 ## unet
 print(f"Loading UNet...")