|
|
@ -20,8 +20,6 @@ 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[0]=="v2" else version
|
|
|
|
|
|
|
|
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"
|
|
|
|
|
|
|
|
|
|
|
@ -77,7 +75,7 @@ from tools.i18n.i18n import I18nAuto
|
|
|
|
|
|
|
|
|
|
|
|
if language != 'auto':
|
|
|
|
if language != 'auto':
|
|
|
|
i18n = I18nAuto(language=language)
|
|
|
|
i18n = I18nAuto(language=language)
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
i18n = I18nAuto()
|
|
|
|
i18n = I18nAuto()
|
|
|
|
|
|
|
|
|
|
|
|
# os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1' # 确保直接启动推理UI时也能够设置。
|
|
|
|
# os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1' # 确保直接启动推理UI时也能够设置。
|
|
|
|