unknown commited on
Commit
ae3efce
·
1 Parent(s): 45be0df

change name make more clear the preetain need path

Browse files
src/f5_tts/train/finetune_cli.py CHANGED
@@ -45,7 +45,7 @@ def parse_args():
45
  parser.add_argument("--save_per_updates", type=int, default=10000, help="Save checkpoint every X steps")
46
  parser.add_argument("--last_per_steps", type=int, default=50000, help="Save last checkpoint every X steps")
47
  parser.add_argument("--finetune", type=bool, default=True, help="Use Finetune")
48
- parser.add_argument("--pretrain", type=str, default=None, help="Use pretrain model for finetune")
49
  parser.add_argument(
50
  "--tokenizer", type=str, default="pinyin", choices=["pinyin", "char", "custom"], help="Tokenizer type"
51
  )
 
45
  parser.add_argument("--save_per_updates", type=int, default=10000, help="Save checkpoint every X steps")
46
  parser.add_argument("--last_per_steps", type=int, default=50000, help="Save last checkpoint every X steps")
47
  parser.add_argument("--finetune", type=bool, default=True, help="Use Finetune")
48
+ parser.add_argument("--pretrain", type=str, default=None, help="the path to the checkpoint")
49
  parser.add_argument(
50
  "--tokenizer", type=str, default="pinyin", choices=["pinyin", "char", "custom"], help="Tokenizer type"
51
  )
src/f5_tts/train/finetune_gradio.py CHANGED
@@ -1331,7 +1331,7 @@ If you encounter a memory error, try reducing the batch size per GPU to a smalle
1331
  with gr.Row():
1332
  ch_finetune = bt_create = gr.Checkbox(label="finetune", value=True)
1333
  tokenizer_file = gr.Textbox(label="Tokenizer File", value="")
1334
- file_checkpoint_train = gr.Textbox(label="Pretrain Model", value="")
1335
 
1336
  with gr.Row():
1337
  exp_name = gr.Radio(label="Model", choices=["F5TTS_Base", "E2TTS_Base"], value="F5TTS_Base")
 
1331
  with gr.Row():
1332
  ch_finetune = bt_create = gr.Checkbox(label="finetune", value=True)
1333
  tokenizer_file = gr.Textbox(label="Tokenizer File", value="")
1334
+ file_checkpoint_train = gr.Textbox(label="Path to the preetrain checkpoint ", value="")
1335
 
1336
  with gr.Row():
1337
  exp_name = gr.Radio(label="Model", choices=["F5TTS_Base", "E2TTS_Base"], value="F5TTS_Base")