attashe commited on
Commit
358ce6d
·
1 Parent(s): 4d4221f

disable e2-tts model

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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, ckpt_path, vocab_file=vocab_path)
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 = load_e2tts() if USING_SPACES else None
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