This commit is contained in:
XXXXRT666 2024-08-03 20:58:28 +08:00
parent 838965b6c2
commit 11850043a8
2 changed files with 8 additions and 4 deletions

View File

@ -597,8 +597,10 @@ def change_choices():
SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights" SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights"
GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights" GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights"
os.makedirs(SoVITS_weight_root, exist_ok=True) os.makedirs("SoVITS_weights",exist_ok=True)
os.makedirs(GPT_weight_root, exist_ok=True) os.makedirs("GPT_weights",exist_ok=True)
os.makedirs("SoVITS_weights_v2",exist_ok=True)
os.makedirs("GPT_weights_v2",exist_ok=True)
def get_weights_names(): def get_weights_names():

View File

@ -124,8 +124,10 @@ def get_weights_names():
return SoVITS_names,GPT_names return SoVITS_names,GPT_names
SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights" SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights"
GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights" GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights"
os.makedirs(SoVITS_weight_root,exist_ok=True) os.makedirs("SoVITS_weights",exist_ok=True)
os.makedirs(GPT_weight_root,exist_ok=True) os.makedirs("GPT_weights",exist_ok=True)
os.makedirs("SoVITS_weights_v2",exist_ok=True)
os.makedirs("GPT_weights_v2",exist_ok=True)
SoVITS_names,GPT_names = get_weights_names() SoVITS_names,GPT_names = get_weights_names()
def custom_sort_key(s): def custom_sort_key(s):