ThreadAbort commited on
Commit
29d19bd
1 Parent(s): fda8dc2
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ model/__pycache__/
2
+
app.py CHANGED
@@ -79,8 +79,8 @@ def load_model(exp_name, model_cls, model_cfg, ckpt_step):
79
  F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
80
  E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
81
 
82
- F5TTS_ema_model, F5TTS_base_model = load_model("F5TTS_Base", DiT, F5TTS_model_cfg, 1200000)
83
- E2TTS_ema_model, E2TTS_base_model = load_model("E2TTS_Base", UNetT, E2TTS_model_cfg, 1200000)
84
 
85
  @spaces.GPU
86
  def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence, progress = gr.Progress()):
 
79
  F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
80
  E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
81
 
82
+ F5TTS_ema_model, F5TTS_base_model = load_model("F5TTS", DiT, F5TTS_model_cfg, 1200000)
83
+ E2TTS_ema_model, E2TTS_base_model = load_model("E2TTS", UNetT, E2TTS_model_cfg, 1200000)
84
 
85
  @spaces.GPU
86
  def infer(ref_audio_orig, ref_text, gen_text, exp_name, remove_silence, progress = gr.Progress()):
model/backbones/__pycache__/dit.cpython-310.pyc ADDED
Binary file (4.57 kB). View file
 
model/backbones/__pycache__/mmdit.cpython-310.pyc ADDED
Binary file (3.94 kB). View file
 
model/backbones/__pycache__/unett.cpython-310.pyc ADDED
Binary file (5.28 kB). View file