Use Hub Python API for model downloads

scripts/bootstrap_comfy.py

@@ -191,30 +191,26 @@ def download_models(dry_run: bool = False) -> None:
     """
 
     token = os.environ.get("HF_TOKEN")
-    token_args = ["--token", token] if token else []
-    hf_cmd = shutil.which("hf") or shutil.which("huggingface-cli") or "hf"
-    run(
-        [
-            hf_cmd,
-            "download",
-            "openbmb/VoxCPM2",
-            "--local-dir",
-            str(model_target("voxcpm2")),
-            *token_args,
-        ],
-        dry_run=dry_run,
+    if dry_run:
+        print(f"+ snapshot_download openbmb/VoxCPM2 -> {model_target('voxcpm2')}")
+        print(
+            "+ hf_hub_download "
+            f"Kijai/MelBandRoFormer_comfy/MelBandRoformer_fp32.safetensors -> {model_target('melband')}"
+        )
+        return
+
+    from huggingface_hub import hf_hub_download, snapshot_download
+
+    snapshot_download(
+        repo_id="openbmb/VoxCPM2",
+        local_dir=model_target("voxcpm2"),
+        token=token,
     )
-    run(
-        [
-            hf_cmd,
-            "download",
-            "Kijai/MelBandRoFormer_comfy",
-            "MelBandRoformer_fp32.safetensors",
-            "--local-dir",
-            str(model_target("melband")),
-            *token_args,
-        ],
-        dry_run=dry_run,
+    hf_hub_download(
+        repo_id="Kijai/MelBandRoFormer_comfy",
+        filename="MelBandRoformer_fp32.safetensors",
+        local_dir=model_target("melband"),
+        token=token,
     )