Merge branch 'main' of https://huggingface.co/spaces/Hyathi/LatentSync

app.py

@@ -5,12 +5,13 @@ from omegaconf import OmegaConf
 import argparse
 from datetime import datetime
 import subprocess
+import os
 
 CONFIG_PATH = Path("configs/unet/second_stage.yaml")
 CHECKPOINT_PATH = Path("checkpoints/latentsync_unet.pt")
 
-subprocess.run(["huggingface-cli", "download", "chunyu-li/LatentSync", "--local-dir", "checkpoints", "--exclude", "*.git*", "README.md"])
-
+# subprocess.run(["huggingface-cli", "download", "Hyathi/LatentSync", "--local-dir", "checkpoints", "--exclude", "*.git*", "README.md"])
+subprocess.run(["huggingface-cli", "download", "Hyathi/LatentSync", "--local-dir", "checkpoints", "--exclude", "*.git*", "README.md", "--token", os.environ["HF_TOKEN"]])
 
 def process_video(
     video_path,

setup_env.sh

@@ -14,7 +14,7 @@ pip install -r requirements.txt
 sudo apt -y install libgl1
 
 # Download all the checkpoints from HuggingFace
-huggingface-cli download chunyu-li/LatentSync --local-dir checkpoints --exclude "*.git*" "README.md"
+huggingface-cli download Hyathi/LatentSync --local-dir checkpoints --exclude "*.git*" "README.md"
 
 # Soft links for the auxiliary models
 mkdir -p ~/.cache/torch/hub/checkpoints