Spaces:
Running
Running
ThreadAbort
commited on
Commit
•
29d19bd
1
Parent(s):
fda8dc2
gitignore
Browse files
.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("
|
83 |
-
E2TTS_ema_model, E2TTS_base_model = load_model("
|
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
|
|