from huggingface_hub import snapshot_download
import os

def download_model(local_dir):
    repo_id = "waytan22/SongGeneration-LeVo"

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

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