mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-12-16 01:06:57 +08:00
Improve G2PWPinyin model handling
- Add 'bert_pretrained_dir' environment variable as the highest priority for model_source - Replace redundant g2pw initialization in chinese2.py with import from download.py - Remove unused 'parent_directory' variable
This commit is contained in:
parent
11aa78bd9b
commit
35fac290ba
@ -7,7 +7,7 @@ from text.g2pw import G2PWPinyin
|
||||
|
||||
g2pw = G2PWPinyin(
|
||||
model_dir="GPT_SoVITS/text/G2PWModel",
|
||||
model_source="GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large",
|
||||
model_source=os.environ.get("bert_pretrained_dir") or os.environ.get("bert_path") or "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large",
|
||||
v_to_u=False,
|
||||
neutral_tone_with_five=True,
|
||||
)
|
||||
|
||||
@ -28,15 +28,8 @@ import jieba_fast.posseg as psg
|
||||
is_g2pw = True # True if is_g2pw_str.lower() == 'true' else False
|
||||
if is_g2pw:
|
||||
# print("当前使用g2pw进行拼音推理")
|
||||
from text.g2pw import G2PWPinyin, correct_pronunciation
|
||||
|
||||
parent_directory = os.path.dirname(current_file_path)
|
||||
g2pw = G2PWPinyin(
|
||||
model_dir="GPT_SoVITS/text/G2PWModel",
|
||||
model_source=os.environ.get("bert_path", "GPT_SoVITS/pretrained_models/chinese-roberta-wwm-ext-large"),
|
||||
v_to_u=False,
|
||||
neutral_tone_with_five=True,
|
||||
)
|
||||
from text.g2pw import correct_pronunciation
|
||||
from download import g2pw
|
||||
|
||||
rep_map = {
|
||||
":": ",",
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user