disable e2-tts model
Browse files
app.py
CHANGED
@@ -60,10 +60,8 @@ vocoder = load_vocoder(vocoder_name="bigvgan")
|
|
60 |
|
61 |
|
62 |
def load_f5tts(ckpt_path=str(cached_path("hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.safetensors"))):
|
63 |
-
ckpt_path = model_file
|
64 |
-
vocab_path = vocab_file
|
65 |
F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
|
66 |
-
return load_model(DiT, F5TTS_model_cfg,
|
67 |
|
68 |
|
69 |
def load_e2tts(ckpt_path=str(cached_path("hf://SWivid/E2-TTS/E2TTS_Base/model_1200000.safetensors"))):
|
@@ -83,7 +81,7 @@ def load_custom(ckpt_path: str, vocab_path="", model_cfg=None):
|
|
83 |
|
84 |
|
85 |
F5TTS_ema_model = load_f5tts()
|
86 |
-
E2TTS_ema_model =
|
87 |
custom_ema_model, pre_custom_path = None, ""
|
88 |
|
89 |
chat_model_state = None
|
|
|
60 |
|
61 |
|
62 |
def load_f5tts(ckpt_path=str(cached_path("hf://SWivid/F5-TTS/F5TTS_Base/model_1200000.safetensors"))):
|
|
|
|
|
63 |
F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
|
64 |
+
return load_model(DiT, F5TTS_model_cfg, model_file, vocab_file=vocab_file)
|
65 |
|
66 |
|
67 |
def load_e2tts(ckpt_path=str(cached_path("hf://SWivid/E2-TTS/E2TTS_Base/model_1200000.safetensors"))):
|
|
|
81 |
|
82 |
|
83 |
F5TTS_ema_model = load_f5tts()
|
84 |
+
E2TTS_ema_model = None
|
85 |
custom_ema_model, pre_custom_path = None, ""
|
86 |
|
87 |
chat_model_state = None
|