root commited on
Commit
4b06a6c
·
1 Parent(s): 45a12c7
Files changed (1) hide show
  1. download.py +2 -0
download.py CHANGED
@@ -1,6 +1,8 @@
1
  from huggingface_hub import snapshot_download
2
  import os
3
 
 
 
4
 
5
  def download_model(local_dir, repo_id="tencent/SongGeneration", revision="aa9d1b3"):
6
  downloaded_path = snapshot_download(
 
1
  from huggingface_hub import snapshot_download
2
  import os
3
 
4
+ os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "1200"
5
+ # os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
6
 
7
  def download_model(local_dir, repo_id="tencent/SongGeneration", revision="aa9d1b3"):
8
  downloaded_path = snapshot_download(