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="
|
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="
|
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")
|