SWivid commited on
Commit
d280126
·
1 Parent(s): 88e3741

minor fix for finetune_gradio.py

Browse files
src/f5_tts/train/finetune_gradio.py CHANGED
@@ -827,7 +827,6 @@ def infer(file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step):
827
  else:
828
  device_test = None
829
 
830
- device_test = "cpu"
831
  if last_checkpoint != file_checkpoint or last_device != device_test:
832
  if last_checkpoint != file_checkpoint:
833
  last_checkpoint = file_checkpoint
 
827
  else:
828
  device_test = None
829
 
 
830
  if last_checkpoint != file_checkpoint or last_device != device_test:
831
  if last_checkpoint != file_checkpoint:
832
  last_checkpoint = file_checkpoint