gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-01-16 06:53:17 -08:00
Code Issues Packages Projects Releases Wiki Activity

fix tts_api after clear in memory (#216)

Browse Source
This commit is contained in:
lpscr
2024-10-22 14:56:16 +03:00
committed by GitHub
parent 608e224f9d
commit 32c3ee7701
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
finetune_gradio.py
View File

@@ -254,6 +254,7 @@ def start_training(
del tts_api
gc.collect()
torch.cuda.empty_cache()
tts_api = None
path_project = os.path.join(path_data, dataset_name + "_pinyin")
@@ -698,7 +699,7 @@ def get_random_sample_infer(project_name):
)
def infer(project_name, file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step):
def infer(file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe_step):
global last_checkpoint, last_device, tts_api
if not os.path.isfile(file_checkpoint):
@@ -917,7 +918,7 @@ with gr.Blocks() as app:
check_button_infer.click(
fn=infer,
inputs=[project_name, file_checkpoint_pt, exp_name, ref_text, ref_audio, gen_text, nfe_step],
inputs=[file_checkpoint_pt, exp_name, ref_text, ref_audio, gen_text, nfe_step],
outputs=[gen_audio],
)