SongGeneration / download.py
rishi2025's picture
Update download.py
1980af7 verified
raw
history blame
438 Bytes
from huggingface_hub import snapshot_download
import os
def download_model(local_dir):
repo_id = "tencent/SongGeneration"
downloaded_path = snapshot_download(
repo_id=repo_id,
local_dir=local_dir,
# revision="0c80d30",
token=os.environ.get("HF_TOKEN"),
ignore_patterns=['.git*']
)
print(f"File downloaded to:{downloaded_path}")
if __name__ == '__main__':
download_model('.')