kevinwang676 commited on
Commit
3b7fc6d
·
verified ·
1 Parent(s): b190815

Update GPT_SoVITS/app.py

Browse files
Files changed (1) hide show
  1. GPT_SoVITS/app.py +2 -2
GPT_SoVITS/app.py CHANGED
@@ -54,7 +54,7 @@ def get_pretrain_model_path(env_name, log_file, def_path):
54
  device = "cuda" if torch.cuda.is_available() else "cpu"
55
 
56
  gpt_path = get_pretrain_model_path('gpt_path', "./gweight.txt",
57
- "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt")
58
 
59
  sovits_path = get_pretrain_model_path('sovits_path', "./sweight.txt",
60
  "GPT_SoVITS/pretrained_models/s2G2333k.pth")
@@ -294,7 +294,7 @@ def change_choices():
294
 
295
 
296
  pretrained_sovits_name = "GPT_SoVITS/pretrained_models/s2G2333k.pth"
297
- pretrained_gpt_name = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
298
  SoVITS_weight_root = "SoVITS_weights"
299
  GPT_weight_root = "GPT_weights"
300
  os.makedirs(SoVITS_weight_root, exist_ok=True)
 
54
  device = "cuda" if torch.cuda.is_available() else "cpu"
55
 
56
  gpt_path = get_pretrain_model_path('gpt_path', "./gweight.txt",
57
+ "GPT_SoVITS/pretrained_models/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt")
58
 
59
  sovits_path = get_pretrain_model_path('sovits_path', "./sweight.txt",
60
  "GPT_SoVITS/pretrained_models/s2G2333k.pth")
 
294
 
295
 
296
  pretrained_sovits_name = "GPT_SoVITS/pretrained_models/s2G2333k.pth"
297
+ pretrained_gpt_name = "GPT_SoVITS/pretrained_models/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt"
298
  SoVITS_weight_root = "SoVITS_weights"
299
  GPT_weight_root = "GPT_weights"
300
  os.makedirs(SoVITS_weight_root, exist_ok=True)