Update GPT_SoVITS/app_vc.py

GPT_SoVITS/app_vc.py

@@ -56,15 +56,11 @@ def get_pretrain_model_path(env_name, log_file, def_path):
 # device = "cuda" if torch.cuda.is_available() else "cpu"
 device = 'cpu'
 
-#gpt_path = get_pretrain_model_path('gpt_path', "./gweight.txt", 
-#                                   "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt")
+gpt_path = get_pretrain_model_path('gpt_path', "./gweight.txt", 
+                                   "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt")
 
-#sovits_path = get_pretrain_model_path('sovits_path', "./sweight.txt",
-#                                      "GPT_SoVITS/pretrained_models/s2G488k.pth")
-
-gpt_path = "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt"
-
-sovits_path = "GPT_SoVITS/pretrained_models/s2G488k.pth"
+sovits_path = get_pretrain_model_path('sovits_path', "./sweight.txt",
+                                      "GPT_SoVITS/pretrained_models/s2G488k.pth")
 
 cnhubert_base_path = get_pretrain_model_path("cnhubert_base_path", '', "GPT_SoVITS/pretrained_models/chinese-hubert-base")
 
@@ -305,8 +301,8 @@ def change_choices():
 
 pretrained_sovits_name = "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth"
 pretrained_gpt_name = "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt"
-SoVITS_weight_root = "SoVITS_weights"
-GPT_weight_root = "GPT_weights"
+SoVITS_weight_root = "SoVITS_weights_v2"
+GPT_weight_root = "GPT_weights_v2"
 os.makedirs(SoVITS_weight_root, exist_ok=True)
 os.makedirs(GPT_weight_root, exist_ok=True)