Upload download_checkpoints.py with huggingface_hub

download_checkpoints.py

@@ -9,6 +9,7 @@ CHECKPOINT_FILES = ["duts_saliency.ckpt", "nyu_depth.ckpt", "pascal_segmentation
 
 def download_all():
     os.makedirs(CHECKPOINT_DIR, exist_ok=True)
+    token = os.environ.get("HF_TOKEN")
     for fname in CHECKPOINT_FILES:
         path = os.path.join(CHECKPOINT_DIR, fname)
         if not os.path.exists(path):
@@ -18,6 +19,7 @@ def download_all():
                 filename=fname,
                 repo_type="model",
                 local_dir=CHECKPOINT_DIR,
+                token=token,
             )
             print(f"[DOWNLOAD] {fname} -> {downloaded}")
         else: