blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-05 22:20:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复在其他软件import时，找不到模型文件

Browse Source
This commit is contained in:
CyberWon 2025-04-25 17:57:22 +08:00
parent 9b72d61ad1
commit 4208e7635c
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
GPT_SoVITS/text/chinese2.py
View File

@ -31,9 +31,12 @@ if is_g2pw:
from text.g2pw import G2PWPinyin, correct_pronunciation
parent_directory = os.path.dirname(current_file_path)
# 防止作为第三方库导入时，找不到模型文件
model_dir = os.path.join(current_file_path, "G2PWModel")
model_source = os.path.join(parent_directory, "pretrained_models/chinese-roberta-wwm-ext-large")
g2pw = G2PWPinyin(
model_dir="GPT_SoVITS/text/G2PWModel",
model_source=os.environ.get("bert_path", "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large"),
model_dir=model_dir,
model_source=os.environ.get("bert_path", model_source),
v_to_u=False,
neutral_tone_with_five=True,
)