From 838965b6c20c537be2da795dc168ac6c8fab010e Mon Sep 17 00:00:00 2001 From: XXXXRT666 Date: Sat, 3 Aug 2024 20:54:59 +0800 Subject: [PATCH] chores --- GPT_SoVITS/inference_webui.py | 4 ++-- webui.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py index b94919fd..31298a79 100644 --- a/GPT_SoVITS/inference_webui.py +++ b/GPT_SoVITS/inference_webui.py @@ -595,8 +595,8 @@ def change_choices(): return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"} -SoVITS_weight_root = f"SoVITS_weights/{version}" -GPT_weight_root = f"GPT_weights/{version}" +SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights" +GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights" os.makedirs(SoVITS_weight_root, exist_ok=True) os.makedirs(GPT_weight_root, exist_ok=True) diff --git a/webui.py b/webui.py index b12ada5a..0df7c941 100644 --- a/webui.py +++ b/webui.py @@ -122,8 +122,8 @@ def get_weights_names(): for name in os.listdir(GPT_weight_root): if name.endswith(".ckpt"): GPT_names.append(name) return SoVITS_names,GPT_names -SoVITS_weight_root=f"SoVITS_weights/{version}" -GPT_weight_root=f"GPT_weights/{version}" +SoVITS_weight_root="SoVITS_weights_v2" if version=='v2' else "SoVITS_weights" +GPT_weight_root="GPT_weights_v2" if version=='v2' else "GPT_weights" os.makedirs(SoVITS_weight_root,exist_ok=True) os.makedirs(GPT_weight_root,exist_ok=True) SoVITS_names,GPT_names = get_weights_names()