diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py index 3046d7af..a40ce860 100644 --- a/GPT_SoVITS/inference_webui.py +++ b/GPT_SoVITS/inference_webui.py @@ -1,22 +1,24 @@ import os,re import pdb - +os.environ["no_proxy"] = "localhost, 127.0.0.1, ::1" 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", "pretrained_models/chinese-hubert-base" + "cnhubert_base_path", "GPT_SoVITS/pretrained_models/chinese-hubert-base" ) 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) is_share = os.environ.get("is_share", "False") is_share=eval(is_share) if "_CUDA_VISIBLE_DEVICES" in os.environ: 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")) import gradio as gr from transformers import AutoModelForMaskedLM, AutoTokenizer