M4xjunior commited on
Commit
375ecba
1 Parent(s): a9dabb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -66,7 +66,7 @@ def load_f5tts():
66
  ckpt_path = hf_hub_download(repo_id=repo_id, filename=filename, use_auth_token=token)
67
 
68
  F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
69
- return load_model(DiT, F5TTS_model_cfg, ckpt_path, use_ema=True)
70
 
71
 
72
 
@@ -148,7 +148,6 @@ def infer(
148
  ema_model,
149
  vocoder,
150
  cross_fade_duration=cross_fade_duration,
151
- nfe_step=64,
152
  speed=speed,
153
  show_info=show_info,
154
  progress=gr.Progress(),
 
66
  ckpt_path = hf_hub_download(repo_id=repo_id, filename=filename, use_auth_token=token)
67
 
68
  F5TTS_model_cfg = dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4)
69
+ return load_model(DiT, F5TTS_model_cfg, ckpt_path)
70
 
71
 
72
 
 
148
  ema_model,
149
  vocoder,
150
  cross_fade_duration=cross_fade_duration,
 
151
  speed=speed,
152
  show_info=show_info,
153
  progress=gr.Progress(),