Real_Time_Interactive_Avatar_v2 / download_models.py
wedyanessam's picture
Update download_models.py
2f8e7ab verified
raw
history blame
773 Bytes
import os
from huggingface_hub import snapshot_download
def maybe_download_model(repo_id, local_dir):
if not os.path.exists(local_dir) or not os.listdir(local_dir):
print(f"📥 Downloading model from {repo_id} ...")
snapshot_download(
repo_id=repo_id,
local_dir=local_dir,
local_dir_use_symlinks=False
)
else:
print(f"✅ Model already exists at {local_dir}, skipping download.")
# تحميل النماذج فقط إذا ما كانت موجودة
maybe_download_model("Wan-AI/Wan2.1-I2V-14B-720P", "./models/Wan2.1-I2V-14B-720P")
maybe_download_model("facebook/wav2vec2-base-960h", "./models/wav2vec2-base-960h")
maybe_download_model("acvlab/FantasyTalking", "./models/fantasytalking_model")