Initial commit

echobox/src/echobox/vc.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 
 import librosa
@@ -10,7 +11,7 @@ from .models.s3gen import S3GEN_SR, S3Gen
 
 
 REPO_ID = "Lookingsoft-team/EchoBoxTTS"
-
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
 class ChatterboxVC:
     ENC_COND_LEN = 6 * S3_SR
@@ -53,7 +54,7 @@ class ChatterboxVC:
     @classmethod
     def from_pretrained(cls, device) -> 'ChatterboxVC':
         for fpath in ["s3gen.pt", "conds.pt"]:
-            local_path = hf_hub_download(repo_id=REPO_ID, filename=fpath)
+            local_path = hf_hub_download(repo_id=REPO_ID, filename=fpath, token=HF_TOKEN)
 
         return cls.from_local(Path(local_path).parent, device)