Update finetune_gradio.py. fix flags

SWivid · Dec 28, 2024 · fa8dab6 · fa8dab6
1 parent 4e8ff53
commit fa8dab6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/f5_tts/train/finetune_gradio.py b/src/f5_tts/train/finetune_gradio.py
@@ -453,20 +453,20 @@ def start_training(
     )
 
     if finetune:
-        cmd += f" --finetune {finetune}"
+        cmd += f" --finetune"
 
     if file_checkpoint_train != "":
         cmd += f" --pretrain {file_checkpoint_train}"
 
     if tokenizer_file != "":
         cmd += f" --tokenizer_path {tokenizer_file}"
 
-    cmd += f" --tokenizer {tokenizer_type} "
+    cmd += f" --tokenizer {tokenizer_type}"
 
-    cmd += f" --log_samples True --logger {logger} "
+    cmd += f" --log_samples --logger {logger}"
 
     if ch_8bit_adam:
-        cmd += " --bnb_optimizer True "
+        cmd += " --bnb_optimizer"
 
     print("run command : \n" + cmd + "\n")