Spaces:
Running
Running
kevinwang676
commited on
Commit
•
ba8ec5d
1
Parent(s):
2331da5
Update GPT_SoVITS/inference_webui.py
Browse files
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
|