Skip to content

Commit

Permalink
Fix for incorrect defaults in the finetune_gradio interface (#755)
Browse files Browse the repository at this point in the history
* Add missing components to setup_load_settings in finetune_gradio
  • Loading branch information
hcsolakoglu authored Jan 29, 2025
1 parent f099649 commit bebbfbb
Showing 1 changed file with 19 additions and 18 deletions.
37 changes: 19 additions & 18 deletions src/f5_tts/train/finetune_gradio.py
Original file line number Diff line number Diff line change
Expand Up @@ -1744,25 +1744,26 @@ def get_audio_select(file_sample):

def setup_load_settings():
output_components = [
exp_name,
learning_rate,
batch_size_per_gpu,
batch_size_type,
max_samples,
grad_accumulation_steps,
max_grad_norm,
epochs,
num_warmup_updates,
save_per_updates,
last_per_updates,
ch_finetune,
file_checkpoint_train,
tokenizer_type,
tokenizer_file,
mixed_precision,
cd_logger,
exp_name, # 1
learning_rate, # 2
batch_size_per_gpu, # 3
batch_size_type, # 4
max_samples, # 5
grad_accumulation_steps, # 6
max_grad_norm, # 7
epochs, # 8
num_warmup_updates, # 9
save_per_updates, # 10
keep_last_n_checkpoints, # 11
last_per_updates, # 12
ch_finetune, # 13
file_checkpoint_train, # 14
tokenizer_type, # 15
tokenizer_file, # 16
mixed_precision, # 17
cd_logger, # 18
ch_8bit_adam, # 19
]

return output_components

outputs = setup_load_settings()
Expand Down

0 comments on commit bebbfbb

Please sign in to comment.