gradio finetune fix wrong value (#301)

* fix wrong value print vocab
This commit is contained in:
lpscr
2024-10-28 11:03:10 +02:00
committed by GitHub
parent a7fd2e7e9a
commit 700039b554
2 changed files with 3 additions and 1 deletions

View File

@@ -106,6 +106,8 @@ def main():
vocab_char_map, vocab_size = get_tokenizer(tokenizer_path, tokenizer)
print("\nvocab : ", vocab_size)
mel_spec_kwargs = dict(
target_sample_rate=target_sample_rate,
n_mel_channels=n_mel_channels,

View File

@@ -1389,7 +1389,7 @@ If you encounter a memory error, try reducing the batch size per GPU to a smalle
save_per_updates.value = save_per_updatesv
last_per_steps.value = last_per_stepsv
ch_finetune.value = finetunev
file_checkpoint_train.value = file_checkpoint_train
file_checkpoint_train.value = file_checkpoint_trainv
tokenizer_type.value = tokenizer_typev
tokenizer_file.value = tokenizer_filev
mixed_precision.value = mixed_precisionv