small updates and to or in model load

lpscr · Oct 21, 2024 · 7920e52 · 7920e52
1 parent 20f823d
commit 7920e52
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/finetune_gradio.py b/finetune_gradio.py
@@ -709,14 +709,16 @@ def infer(project_name, file_checkpoint, exp_name, ref_text, ref_audio, gen_text
     else:
         device_test = None
 
-    if last_checkpoint != file_checkpoint and last_device != device_test:
+    if last_checkpoint != file_checkpoint or last_device != device_test:
         if last_checkpoint != file_checkpoint:
             last_checkpoint = file_checkpoint
         if last_device != device_test:
             last_device = device_test
 
         tts_api = F5TTS(model_type=exp_name, ckpt_file=file_checkpoint, device=device_test)
 
+        print("update", device_test, file_checkpoint)
+
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
         tts_api.infer(gen_text=gen_text, ref_text=ref_text, ref_file=ref_audio, nfe_step=nfe_step, file_wave=f.name)
         return f.name