gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-07-28 13:26:59 -07:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #751 from hcsolakoglu/fix-finetune-gradio-dropdown

Browse Source 
Small fix for the checkpoint dropdown menu in finetune gradio
This commit is contained in:
Yushen CHEN
2025-01-28 22:52:25 +08:00
committed by GitHub
parent 607b92b391 f8cc2446c8
commit 738d502f3b
1 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/f5_tts/train/finetune_gradio.py
+15 -5
View File
@@ -1261,12 +1261,22 @@ def get_checkpoints_project(project_name, is_gradio=True):
if os.path.isdir(path_project_ckpts):
files_checkpoints = glob(os.path.join(path_project_ckpts, project_name, "*.pt"))
files_checkpoints = sorted(
files_checkpoints,
key=lambda x: int(os.path.basename(x).split("_")[1].split(".")[0])
if os.path.basename(x) != "model_last.pt"
else float("inf"),
# Separate pretrained and regular checkpoints
pretrained_checkpoints = [f for f in files_checkpoints if "pretrained_" in os.path.basename(f)]
regular_checkpoints = [
f
for f in files_checkpoints
if "pretrained_" not in os.path.basename(f) and "model_last.pt" not in os.path.basename(f)
]
last_checkpoint = [f for f in files_checkpoints if "model_last.pt" in os.path.basename(f)]
# Sort regular checkpoints by number
regular_checkpoints = sorted(
regular_checkpoints, key=lambda x: int(os.path.basename(x).split("_")[1].split(".")[0])
)
# Combine in order: pretrained, regular, last
files_checkpoints = pretrained_checkpoints + regular_checkpoints + last_checkpoint
else:
files_checkpoints = []