diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py index 31298a79..a7e775e7 100644 --- a/GPT_SoVITS/inference_webui.py +++ b/GPT_SoVITS/inference_webui.py @@ -597,8 +597,10 @@ def change_choices(): 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) +os.makedirs("SoVITS_weights",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(): diff --git a/webui.py b/webui.py index 0df7c941..faebb784 100644 --- a/webui.py +++ b/webui.py @@ -124,8 +124,10 @@ def get_weights_names(): return SoVITS_names,GPT_names 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) +os.makedirs("SoVITS_weights",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() def custom_sort_key(s):