Gregniuki commited on
Commit
2691558
·
verified ·
1 Parent(s): 774f8ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -108,19 +108,17 @@ chat_tokenizer_state = None
108
 
109
 
110
  # load models
111
- F5TTS_model_cfg = dict(
112
- dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4
113
- )
114
  E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
115
 
116
  F5TTS_ema_model = load_custom(
117
- "https://huggingface.co/Gregniuki/F5-tts_English_German_Polish/resolve/main/English/model_222600.pt", "", F5TTS_model_cfg
118
  )
119
  E2TTS_ema_model = load_custom(
120
- "https://huggingface.co/Gregniuki/F5-tts_English_German_Polish/resolve/main/Polish2/model_1200000.pt", "", F5TTS_model_cfg
121
  )
122
  E2TTS_ema_model2 = load_custom(
123
- "https://huggingface.co/Gregniuki/F5-tts_English_German_Polish/resolve/main/Polish/model_500000.pt", "", F5TTS_model_cfg
124
  )
125
 
126
  def chunk_text(text, max_chars=135):
 
108
 
109
 
110
  # load models
111
+ F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
 
 
112
  E2TTS_model_cfg = dict(dim=1024, depth=24, heads=16, ff_mult=4)
113
 
114
  F5TTS_ema_model = load_custom(
115
+ "hf://Gregniuki/F5-tts_English_German_Polish/English/model_222600.pt", "", F5TTS_model_cfg
116
  )
117
  E2TTS_ema_model = load_custom(
118
+ "hf://Gregniuki/F5-tts_English_German_Polish/Polish2/model_1200000.pt", "", F5TTS_model_cfg
119
  )
120
  E2TTS_ema_model2 = load_custom(
121
+ "hf://Gregniuki/F5-tts_English_German_Polish/Polish/model_500000.pt", "", F5TTS_model_cfg
122
  )
123
 
124
  def chunk_text(text, max_chars=135):