From bbe0d4d3eb546c183c85bc97dcb9a354fda9052c Mon Sep 17 00:00:00 2001
From: unknown <lazaros2009a@hotmail.com>
Date: Wed, 6 Nov 2024 00:23:51 +0200
Subject: [PATCH] fix value pretrain

---
 src/f5_tts/train/finetune_gradio.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/f5_tts/train/finetune_gradio.py b/src/f5_tts/train/finetune_gradio.py
index 4b65f5d3b..166acb1ad 100644
--- a/src/f5_tts/train/finetune_gradio.py
+++ b/src/f5_tts/train/finetune_gradio.py
@@ -459,7 +459,7 @@ def start_training(
         cmd += f" --finetune {finetune}"
 
     if file_checkpoint_train != "":
-        cmd += f" --file_checkpoint_train {file_checkpoint_train}"
+        cmd += f" --pretrain {file_checkpoint_train}"
 
     if tokenizer_file != "":
         cmd += f" --tokenizer_path {tokenizer_file}"