Spaces:
Runtime error
Runtime error
File size: 1,817 Bytes
6b25f66 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import os
from huggingface_hub import snapshot_download
def download_files_from_repo():
# check the last ckpts are downloaded
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, "SMPLer-X/pretrained_models/smpler_x_s32.pth.tar")
if os.path.exists(last_ckpt_path):
return
else:
# separate the download of the files for better debugging
snapshot_download(
repo_id=repo_id, local_dir=local_dir, repo_type="space", allow_patterns="frame-interpolation-pytorch/*.pt", force_download=True
)
snapshot_download(
repo_id=repo_id, local_dir=local_dir, repo_type="space", allow_patterns="Wav2Lip/checkpoints/*.pth", force_download=True
)
snapshot_download(
repo_id=repo_id, local_dir=local_dir, repo_type="space", allow_patterns="datasets/cached_ckpts/*", force_download=True
)
snapshot_download(
repo_id=repo_id, local_dir=local_dir, repo_type="space", allow_patterns="datasets/cached_graph/*", force_download=True
)
snapshot_download(
repo_id=repo_id, local_dir=local_dir, repo_type="space", allow_patterns="emage/smplx_models/smplx/*", force_download=True
)
snapshot_download(
repo_id="caizhongang/SMPLer-X",
local_dir=os.path.join(local_dir, "./SMPLer-X"),
repo_type="space",
ignore_patterns="pretrained_models/*",
force_download=True,
)
snapshot_download(
repo_id=repo_id, local_dir=local_dir, repo_type="space", allow_patterns="SMPLer-X/pretrained_models/*", force_download=True
)
print("Downloaded all the necessary files from the repo.")
|