Spaces:
Sleeping
Sleeping
fix: Name change of file.
Browse files
src/chatterbox/mtl_tts.py
CHANGED
|
@@ -147,7 +147,7 @@ class ChatterboxMultilingualTTS:
|
|
| 147 |
ve.to(device).eval()
|
| 148 |
|
| 149 |
t3 = T3(T3Config.multilingual())
|
| 150 |
-
t3_state = load_safetensors(ckpt_dir / "
|
| 151 |
if "model" in t3_state.keys():
|
| 152 |
t3_state = t3_state["model"][0]
|
| 153 |
t3.load_state_dict(t3_state)
|
|
@@ -176,7 +176,7 @@ class ChatterboxMultilingualTTS:
|
|
| 176 |
repo_id=REPO_ID,
|
| 177 |
repo_type="model",
|
| 178 |
revision="main",
|
| 179 |
-
allow_patterns=["ve.pt", "
|
| 180 |
token=os.getenv("HF_TOKEN"),
|
| 181 |
)
|
| 182 |
)
|
|
|
|
| 147 |
ve.to(device).eval()
|
| 148 |
|
| 149 |
t3 = T3(T3Config.multilingual())
|
| 150 |
+
t3_state = load_safetensors(ckpt_dir / "t3_23lang.safetensors", device=str(device))
|
| 151 |
if "model" in t3_state.keys():
|
| 152 |
t3_state = t3_state["model"][0]
|
| 153 |
t3.load_state_dict(t3_state)
|
|
|
|
| 176 |
repo_id=REPO_ID,
|
| 177 |
repo_type="model",
|
| 178 |
revision="main",
|
| 179 |
+
allow_patterns=["ve.pt", "t3_23lang.safetensors", "s3gen.pt", "grapheme_mtl_merged_expanded_v1.json", "conds.pt", "Cangjie5_TC.json"],
|
| 180 |
token=os.getenv("HF_TOKEN"),
|
| 181 |
)
|
| 182 |
)
|