SongGeneration / download.py
root
fix bug
4846f0c
raw
history blame
595 Bytes
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*'],
endpoint="https://hf-mirror.com"
)
print(f"File downloaded to:{downloaded_path}")
if __name__ == '__main__':
download_model('.')