Skip to content

Commit bebbfbb

Browse files
authored
Fix for incorrect defaults in the finetune_gradio interface (#755)
* Add missing components to setup_load_settings in finetune_gradio
1 parent f099649 commit bebbfbb

File tree

1 file changed

+19
-18
lines changed

1 file changed

+19
-18
lines changed

src/f5_tts/train/finetune_gradio.py

Lines changed: 19 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1744,25 +1744,26 @@ def get_audio_select(file_sample):
17441744

17451745
def setup_load_settings():
17461746
output_components = [
1747-
exp_name,
1748-
learning_rate,
1749-
batch_size_per_gpu,
1750-
batch_size_type,
1751-
max_samples,
1752-
grad_accumulation_steps,
1753-
max_grad_norm,
1754-
epochs,
1755-
num_warmup_updates,
1756-
save_per_updates,
1757-
last_per_updates,
1758-
ch_finetune,
1759-
file_checkpoint_train,
1760-
tokenizer_type,
1761-
tokenizer_file,
1762-
mixed_precision,
1763-
cd_logger,
1747+
exp_name, # 1
1748+
learning_rate, # 2
1749+
batch_size_per_gpu, # 3
1750+
batch_size_type, # 4
1751+
max_samples, # 5
1752+
grad_accumulation_steps, # 6
1753+
max_grad_norm, # 7
1754+
epochs, # 8
1755+
num_warmup_updates, # 9
1756+
save_per_updates, # 10
1757+
keep_last_n_checkpoints, # 11
1758+
last_per_updates, # 12
1759+
ch_finetune, # 13
1760+
file_checkpoint_train, # 14
1761+
tokenizer_type, # 15
1762+
tokenizer_file, # 16
1763+
mixed_precision, # 17
1764+
cd_logger, # 18
1765+
ch_8bit_adam, # 19
17641766
]
1765-
17661767
return output_components
17671768

17681769
outputs = setup_load_settings()

0 commit comments

Comments
 (0)