Gregniuki commited on
Commit
bb30bbf
·
verified ·
1 Parent(s): 3528a93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- 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):
 
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):