blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 19:41:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug修复

Browse Source
This commit is contained in:
Downupanddownup 2024-04-29 00:20:13 +08:00
parent d6e255a071
commit b1ad8b5dcd
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Ref_Audio_Selector/common/model_manager.py
View File

@ -19,16 +19,16 @@ def custom_sort_key(s):
def get_gpt_model_names():
gpt_names = [pretrained_gpt_name]
for name in os.listdir(SoVITS_weight_root):
if name.endswith(".ckpt"): gpt_names.append("%s/%s" % (SoVITS_weight_root, name))
for name in os.listdir(GPT_weight_root):
if name.endswith(".ckpt"): gpt_names.append("%s/%s" % (GPT_weight_root, name))
sorted(gpt_names, key=custom_sort_key)
return gpt_names
def get_sovits_model_names():
sovits_names = [pretrained_sovits_name]
for name in os.listdir(GPT_weight_root):
if name.endswith(".pth"): sovits_names.append("%s/%s" % (GPT_weight_root, name))
for name in os.listdir(SoVITS_weight_root):
if name.endswith(".pth"): sovits_names.append("%s/%s" % (SoVITS_weight_root, name))
sorted(sovits_names, key=custom_sort_key)
return sovits_names