| import os |
| from huggingface_hub import snapshot_download, hf_hub_download |
|
|
|
|
| def download_files_from_repo(): |
| |
| repo_id = "H-Liu1997/TANGO" |
| local_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "../") |
| last_ckpt_path = os.path.join(local_dir, "datasets/cached_graph/youtube_test/speaker9.pkl") |
| if os.path.exists(last_ckpt_path): |
| return |
| else: |
| snapshot_download( |
| repo_id="H-Liu1997/tango_cached_utils", local_dir=local_dir, repo_type="dataset", ignore_patterns="datasets/cached_graph/*", force_download=True |
| ) |
| snapshot_download( |
| repo_id="H-Liu1997/tango_cached_utils", local_dir=local_dir, repo_type="dataset", allow_patterns="datasets/cached_graph/youtube_test/speaker9.pkl", force_download=True |
| ) |
| print("Downloaded all the necessary files from the repo.") |
|
|
