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):
|