blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 15:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

chores

Browse Source
This commit is contained in:
XXXXRT666 2024-08-03 12:49:33 +08:00
parent 60caceff5a
commit 7bd3766eb0
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
GPT_SoVITS/inference_webui.py
View File

@ -21,6 +21,7 @@ import torch
version=os.environ.get("version","v1") version=os.environ.get("version","v1")
language=os.environ.get("language","auto") language=os.environ.get("language","auto")
version="v2"if sys.argv[1]=="v2" else version version="v2"if sys.argv[1]=="v2" else version
os.environ['version']=version
language=sys.argv[-1] if sys.argv[-1]!='v2' and sys.argv[-1]!='v1' else language language=sys.argv[-1] if sys.argv[-1]!='v2' and sys.argv[-1]!='v1' else language
pretrained_sovits_name="GPT_SoVITS/pretrained_models/s2G488k.pth"if version=="v1"else"GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth" pretrained_sovits_name="GPT_SoVITS/pretrained_models/s2G488k.pth"if version=="v1"else"GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s2G2333k.pth"
pretrained_gpt_name="GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"if version=="v1"else "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt" pretrained_gpt_name="GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"if version=="v1"else "GPT_SoVITS/pretrained_models/gsv-v2final-pretrained/s1bert25hz-5kh-longer-epoch=12-step=369668.ckpt"
@ -31,18 +32,13 @@ if os.path.exists("./gweight.txt"):
gpt_path = os.environ.get( gpt_path = os.environ.get(
"gpt_path", gweight_data) "gpt_path", gweight_data)
if version == 'v2': if version == 'v2':
gpt_path = pretrained_gpt_name gpt_path = os.environ.get("gpt_path", pretrained_gpt_name)
else:
gpt_path = os.environ.get(
"gpt_path", pretrained_gpt_name)
if os.path.exists("./sweight.txt"): if os.path.exists("./sweight.txt"):
with open("./sweight.txt", 'r', encoding="utf-8") as file: with open("./sweight.txt", 'r', encoding="utf-8") as file:
sweight_data = file.read() sweight_data = file.read()
sovits_path = os.environ.get("sovits_path", sweight_data) sovits_path = os.environ.get("sovits_path", sweight_data)
if version == 'v2': if version == 'v2':
sovits_path = pretrained_sovits_name
else:
sovits_path = os.environ.get("sovits_path", pretrained_sovits_name) sovits_path = os.environ.get("sovits_path", pretrained_sovits_name)
# gpt_path = os.environ.get( # gpt_path = os.environ.get(
# "gpt_path", pretrained_gpt_name # "gpt_path", pretrained_gpt_name