Real_Time_Interactive_Avatar_v2 / download_models.py
wedyanessam's picture
Update download_models.py
c4bb4e4 verified
raw
history blame
570 Bytes
import os
from huggingface_hub import snapshot_download
def maybe_download_model(repo_id, local_dir):
if not os.path.exists(local_dir):
print(f"Downloading {repo_id} into {local_dir}")
snapshot_download(repo_id=repo_id, local_dir=local_dir, local_dir_use_symlinks=False)
else:
print(f"βœ… {repo_id} already downloaded to {local_dir}")
if __name__ == "__main__":
maybe_download_model("wedyanessam/fantasy-talking-model2", "./models/fantasytalking_model")
maybe_download_model("guoyww/wan-video", "./models/Wan2.1-I2V-14B-720P")