Initial commit

echobox/src/echobox/tts.py

@@ -1,3 +1,4 @@
+import os
 from dataclasses import dataclass
 from pathlib import Path
 
@@ -16,7 +17,7 @@ from .models.t3.modules.cond_enc import T3Cond
 
 
 REPO_ID = "Lookingsoft-team/EchoBoxTTS"
-
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
 def punc_norm(text: str) -> str:
     """
@@ -123,7 +124,7 @@ class EchoBoxTTS:
         self.watermarker = perth.PerthImplicitWatermarker()
 
     @classmethod
-    def from_local(cls, ckpt_dir, device) -> 'ChatterboxTTS':
+    def from_local(cls, ckpt_dir, device) -> 'EchoBoxTTS':
         ckpt_dir = Path(ckpt_dir)
 
         ve = VoiceEncoder()
@@ -156,9 +157,9 @@ class EchoBoxTTS:
         return cls(t3, s3gen, ve, tokenizer, device, conds=conds)
 
     @classmethod
-    def from_pretrained(cls, device) -> 'ChatterboxTTS':
+    def from_pretrained(cls, device) -> 'EchoBoxTTS':
         for fpath in ["ve.pt", "t3_cfg.pt", "s3gen.pt", "tokenizer.json", "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)