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

run inference_webui directly

Browse Source
This commit is contained in:
林北1941783147 2024-01-23 22:52:15 +08:00
parent 9a7eabb3e5
commit ebe1480162
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
GPT_SoVITS/inference_webui.py
View File

@ -1,22 +1,24 @@
import os,re import os,re
import pdb import pdb
os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1"
gpt_path = os.environ.get( gpt_path = os.environ.get(
"gpt_path", "pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt" "gpt_path", "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
) )
sovits_path = os.environ.get("sovits_path", "pretrained_models/s2G488k.pth") sovits_path = os.environ.get("sovits_path", "GPT_SoVITS/pretrained_models/s2G488k.pth")
cnhubert_base_path = os.environ.get( cnhubert_base_path = os.environ.get(
"cnhubert_base_path", "pretrained_models/chinese-hubert-base" "cnhubert_base_path", "GPT_SoVITS/pretrained_models/chinese-hubert-base"
) )
bert_path = os.environ.get( bert_path = os.environ.get(
"bert_path", "pretrained_models/chinese-roberta-wwm-ext-large" "bert_path", "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large"
) )
infer_ttswebui = os.environ.get("infer_ttswebui", 9872) infer_ttswebui = os.environ.get("infer_ttswebui", "9872")
infer_ttswebui = int(infer_ttswebui) infer_ttswebui = int(infer_ttswebui)
is_share = os.environ.get("is_share", "False") is_share = os.environ.get("is_share", "False")
is_share=eval(is_share) is_share=eval(is_share)
if "_CUDA_VISIBLE_DEVICES" in os.environ: if "_CUDA_VISIBLE_DEVICES" in os.environ:
os.environ["CUDA_VISIBLE_DEVICES"] = os.environ["_CUDA_VISIBLE_DEVICES"] os.environ["CUDA_VISIBLE_DEVICES"] = os.environ["_CUDA_VISIBLE_DEVICES"]
else:
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
is_half = eval(os.environ.get("is_half", "True")) is_half = eval(os.environ.get("is_half", "True"))
import gradio as gr import gradio as gr
from transformers import AutoModelForMaskedLM, AutoTokenizer from transformers import AutoModelForMaskedLM, AutoTokenizer