gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-12 15:50:07 -08:00
Code Issues Packages Projects Releases Wiki Activity

0.4.4 fix hard coded stdout for finetune-gradio gui

Browse Source
This commit is contained in:
unknown
2025-01-28 11:39:54 +08:00
parent d1f708d442
commit ee2b77064e
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
[project]
name = "f5-tts"
version = "0.4.3"
version = "0.4.4"
description = "F5-TTS: A Fairytaler that Fakes Fluent and Faithful Speech with Flow Matching"
readme = "README.md"
license = {text = "MIT License"}

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

@@ -544,7 +544,7 @@ def start_training(
output = stdout_queue.get_nowait()
print(output, end="")
match = re.search(
r"Epoch (\d+)/(\d+):\s+(\d+)%\|.*\[(\d+:\d+)<.*?loss=(\d+\.\d+), step=(\d+)", output
r"Epoch (\d+)/(\d+):\s+(\d+)%\|.*\[(\d+:\d+)<.*?loss=(\d+\.\d+), update=(\d+)", output
)
if match:
current_epoch = match.group(1)
@@ -552,13 +552,13 @@ def start_training(
percent_complete = match.group(3)
elapsed_time = match.group(4)
loss = match.group(5)
current_step = match.group(6)
current_update = match.group(6)
message = (
f"Epoch: {current_epoch}/{total_epochs}, "
f"Progress: {percent_complete}%, "
f"Elapsed Time: {elapsed_time}, "
f"Loss: {loss}, "
f"Step: {current_step}"
f"Update: {current_update}"
)
yield message, gr.update(interactive=False), gr.update(interactive=True)
elif output.strip():