blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

support gpt-sovits v2

Browse Source 
support gpt-sovits v2
This commit is contained in:
RVC-Boss 2024-08-02 15:43:27 +08:00 committed by GitHub
parent ff6c193f6f
commit 05028e3587
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
GPT_SoVITS/inference_webui.py
View File

@ -6,8 +6,7 @@
全部按英文识别 全部按英文识别
全部按日文识别 全部按日文识别
''' '''
import os, re, logging import logging
import LangSegment
logging.getLogger("markdown_it").setLevel(logging.ERROR) logging.getLogger("markdown_it").setLevel(logging.ERROR)
logging.getLogger("urllib3").setLevel(logging.ERROR) logging.getLogger("urllib3").setLevel(logging.ERROR)
logging.getLogger("httpcore").setLevel(logging.ERROR) logging.getLogger("httpcore").setLevel(logging.ERROR)
@ -15,9 +14,14 @@ logging.getLogger("httpx").setLevel(logging.ERROR)
logging.getLogger("asyncio").setLevel(logging.ERROR) logging.getLogger("asyncio").setLevel(logging.ERROR)
logging.getLogger("charset_normalizer").setLevel(logging.ERROR) logging.getLogger("charset_normalizer").setLevel(logging.ERROR)
logging.getLogger("torchaudio._extension").setLevel(logging.ERROR) logging.getLogger("torchaudio._extension").setLevel(logging.ERROR)
import LangSegment,os, re
import pdb import pdb
import torch import torch
version=os.environ.get("version","v1")
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"
if os.path.exists("./gweight.txt"): if os.path.exists("./gweight.txt"):
with open("./gweight.txt", 'r', encoding="utf-8") as file: with open("./gweight.txt", 'r', encoding="utf-8") as file:
gweight_data = file.read() gweight_data = file.read()
@ -25,18 +29,18 @@ if os.path.exists("./gweight.txt"):
"gpt_path", gweight_data) "gpt_path", gweight_data)
else: else:
gpt_path = os.environ.get( gpt_path = os.environ.get(
"gpt_path", "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt") "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)
else: else:
sovits_path = os.environ.get("sovits_path", "GPT_SoVITS/pretrained_models/s2G488k.pth") sovits_path = os.environ.get("sovits_path", pretrained_sovits_name)
# gpt_path = os.environ.get( # gpt_path = os.environ.get(
# "gpt_path", "pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt" # "gpt_path", pretrained_gpt_name
# ) # )
# sovits_path = os.environ.get("sovits_path", "pretrained_models/s2G488k.pth") # sovits_path = os.environ.get("sovits_path", pretrained_sovits_name)
cnhubert_base_path = os.environ.get( cnhubert_base_path = os.environ.get(
"cnhubert_base_path", "GPT_SoVITS/pretrained_models/chinese-hubert-base" "cnhubert_base_path", "GPT_SoVITS/pretrained_models/chinese-hubert-base"
) )
@ -562,8 +566,6 @@ def change_choices():
return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"} return {"choices": sorted(SoVITS_names, key=custom_sort_key), "__type__": "update"}, {"choices": sorted(GPT_names, key=custom_sort_key), "__type__": "update"}
pretrained_sovits_name = "GPT_SoVITS/pretrained_models/s2G488k.pth"
pretrained_gpt_name = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=50232.ckpt"
SoVITS_weight_root = "SoVITS_weights" SoVITS_weight_root = "SoVITS_weights"
GPT_weight_root = "GPT_weights" GPT_weight_root = "GPT_weights"
os.makedirs(SoVITS_weight_root, exist_ok=True) os.makedirs(SoVITS_weight_root, exist_ok=True)