kevinwang676 commited on
Commit
ba8ec5d
1 Parent(s): 2331da5

Update GPT_SoVITS/inference_webui.py

Browse files
Files changed (1) hide show
  1. GPT_SoVITS/inference_webui.py +2 -2
GPT_SoVITS/inference_webui.py CHANGED
@@ -600,10 +600,10 @@ os.makedirs(GPT_weight_root, exist_ok=True)
600
 
601
 
602
  def get_weights_names():
603
- SoVITS_names = [pretrained_sovits_name]
604
  for name in os.listdir(SoVITS_weight_root):
605
  if name.endswith(".pth"): SoVITS_names.append("%s/%s" % (SoVITS_weight_root, name))
606
- GPT_names = [pretrained_gpt_name]
607
  for name in os.listdir(GPT_weight_root):
608
  if name.endswith(".ckpt"): GPT_names.append("%s/%s" % (GPT_weight_root, name))
609
  return SoVITS_names, GPT_names
 
600
 
601
 
602
  def get_weights_names():
603
+ SoVITS_names = [pretrained_sovits_name, "SoVITS_weights/jay_e10_s60.pth", "SoVITS_weights/linghua_e10_s140.pth"]
604
  for name in os.listdir(SoVITS_weight_root):
605
  if name.endswith(".pth"): SoVITS_names.append("%s/%s" % (SoVITS_weight_root, name))
606
+ GPT_names = [pretrained_gpt_name, "GPT_weights/jay-e15.ckpt", "GPT_weights/linghua-e15.ckpt"]
607
  for name in os.listdir(GPT_weight_root):
608
  if name.endswith(".ckpt"): GPT_names.append("%s/%s" % (GPT_weight_root, name))
609
  return SoVITS_names, GPT_names