gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-01-09 03:43:19 -08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #308 from lpscr/main

Browse Source 
finetune quick fix for the vocab take the vocab from the project
This commit is contained in:
jpgallegoar
2024-10-28 20:37:11 +01:00
committed by GitHub
parent 5d8180ea58 2dddb10c36
commit 6d044f10b0
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/f5_tts/train/finetune_gradio.py
View File

@@ -1161,7 +1161,7 @@ def get_random_sample_infer(project_name):
)
def infer(file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step, use_ema):
def infer(project, file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step, use_ema):
global last_checkpoint, last_device, tts_api, last_ema
if not os.path.isfile(file_checkpoint):
@@ -1182,7 +1182,11 @@ def infer(file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step, us
if last_ema != use_ema:
last_ema = use_ema
tts_api = F5TTS(model_type=exp_name, ckpt_file=file_checkpoint, device=device_test, use_ema=use_ema)
vocab_file = os.path.join(path_data, project, "vocab.txt")
tts_api = F5TTS(
model_type=exp_name, ckpt_file=file_checkpoint, vocab_file=vocab_file, device=device_test, use_ema=use_ema
)
print("update >> ", device_test, file_checkpoint, use_ema)
@@ -1630,7 +1634,7 @@ SOS : check the use_ema setting (True or False) for your model to see what works
check_button_infer.click(
fn=infer,
inputs=[cm_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step, ch_use_ema],
inputs=[cm_project, cm_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step, ch_use_ema],
outputs=[gen_audio, txt_info_gpu],
)