File size: 553 Bytes
2b516e0
 
 
4b06a6c
 
2b516e0
f919478
2b516e0
 
 
3ef9463
2b516e0
 
 
 
 
3ef9463
2b516e0
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from huggingface_hub import snapshot_download
import os

os.environ["HF_HUB_DOWNLOAD_TIMEOUT"] = "1200"
# os.environ["HF_ENDPOINT"] = "https://hf-mirror.com" 

def download_model(local_dir, repo_id="tencent/SongGeneration", revision="aa9d1b3"):
    downloaded_path = snapshot_download(
        repo_id=repo_id,
        local_dir=local_dir,
        revision=revision,
        token=os.environ.get("HF_TOKEN"), 
        ignore_patterns=['.git*']
    )
    print(f"File downloaded to:{downloaded_path}")


if __name__ == '__main__':
    download_model('.')