Update app.py

app.py

@@ -79,8 +79,9 @@ def load_model(page_name, repo_name, exp_name, model_cls, model_cfg, ckpt_step):
         ),
         vocab_char_map=vocab_char_map,
     ).to(device)
+    dtype = None
 
-    model = load_checkpoint(model, ckpt_path, device, use_ema = True)
+    model = load_checkpoint(model, ckpt_path, device, dtype=dtype, use_ema = True)
 
     return model
 
@@ -100,9 +101,9 @@ E2TTS_ema_model = load_model(
 E2TTS_ema_model2 = load_model(
     "Gregniuki", "F5-tts_English_German_Polish", "Polish", DiT, F5TTS_model_cfg, 500000
 )
-# E2TTS_ema_model3 = load_model(
-#     "SWivid", "F5-TTS", "F5TTS_Base", DiT, F5TTS_model_cfg, 1200000
-# )
+ E2TTS_ema_model3 = load_model(
+     "SWivid", "F5-TTS", "F5TTS_Base", DiT, F5TTS_model_cfg, 1200000
+ )
 E2TTS_ema_model4 = load_model(
     "SWivid", "E2-TTS", "E2TTS_Base", UNetT, E2TTS_model_cfg, 1200000
 )