saattrupdan commited on
Commit
d24447c
·
verified ·
1 Parent(s): 3e87669

fix: Name change of file.

Browse files
Files changed (1) hide show
  1. src/chatterbox/mtl_tts.py +2 -2
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 / "t3_mtl23ls_v2.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,7 +176,7 @@ class ChatterboxMultilingualTTS:
176
  repo_id=REPO_ID,
177
  repo_type="model",
178
  revision="main",
179
- allow_patterns=["ve.pt", "t3_mtl23ls_v2.safetensors", "s3gen.pt", "grapheme_mtl_merged_expanded_v1.json", "conds.pt", "Cangjie5_TC.json"],
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
  )