YanTianlong commited on
Commit
3f95f3e
·
1 Parent(s): 41f27dc

Use hf CLI for model downloads

Browse files
Files changed (1) hide show
  1. scripts/bootstrap_comfy.py +3 -2
scripts/bootstrap_comfy.py CHANGED
@@ -192,9 +192,10 @@ def download_models(dry_run: bool = False) -> None:
192
 
193
  token = os.environ.get("HF_TOKEN")
194
  token_args = ["--token", token] if token else []
 
195
  run(
196
  [
197
- "huggingface-cli",
198
  "download",
199
  "openbmb/VoxCPM2",
200
  "--local-dir",
@@ -205,7 +206,7 @@ def download_models(dry_run: bool = False) -> None:
205
  )
206
  run(
207
  [
208
- "huggingface-cli",
209
  "download",
210
  "Kijai/MelBandRoFormer_comfy",
211
  "MelBandRoformer_fp32.safetensors",
 
192
 
193
  token = os.environ.get("HF_TOKEN")
194
  token_args = ["--token", token] if token else []
195
+ hf_cmd = shutil.which("hf") or shutil.which("huggingface-cli") or "hf"
196
  run(
197
  [
198
+ hf_cmd,
199
  "download",
200
  "openbmb/VoxCPM2",
201
  "--local-dir",
 
206
  )
207
  run(
208
  [
209
+ hf_cmd,
210
  "download",
211
  "Kijai/MelBandRoFormer_comfy",
212
  "MelBandRoformer_fp32.safetensors",