Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -96,14 +96,14 @@ tts_model_choice = DEFAULT_TTS_MODEL
|
|
96 |
|
97 |
|
98 |
|
99 |
-
def load_f5tts(ckpt_path=str(cached_path("hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.safetensors"))):
|
100 |
-
|
101 |
-
|
102 |
|
103 |
|
104 |
-
def load_e2tts(ckpt_path=str(cached_path("hf://SWivid/E2-TTS/E2TTS_Base/model_1200000.safetensors"))):
|
105 |
-
E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
|
106 |
-
return load_model(UNetT, E2TTS_model_cfg, ckpt_path)
|
107 |
|
108 |
|
109 |
def load_custom(ckpt_path: str, vocab_path="", model_cfg=None):
|
|
|
96 |
|
97 |
|
98 |
|
99 |
+
#def load_f5tts(ckpt_path=str(cached_path("hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.safetensors"))):
|
100 |
+
# F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
|
101 |
+
# return load_model(DiT, F5TTS_model_cfg, ckpt_path)
|
102 |
|
103 |
|
104 |
+
#def load_e2tts(ckpt_path=str(cached_path("hf://SWivid/E2-TTS/E2TTS_Base/model_1200000.safetensors"))):
|
105 |
+
# E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
|
106 |
+
# return load_model(UNetT, E2TTS_model_cfg, ckpt_path)
|
107 |
|
108 |
|
109 |
def load_custom(ckpt_path: str, vocab_path="", model_cfg=None):
|