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

Repair detail (#1411)

Browse Source 
* Load lang module on demand

* Turn off onnx warning
This commit is contained in:
KamioRinn 2024-08-06 18:26:58 +08:00 committed by GitHub
parent bb224a73c8
commit 2196b2bfc6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
GPT_SoVITS/feature_extractor/cnhubert.py
View File

@ -4,8 +4,10 @@ import librosa
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import soundfile as sf import soundfile as sf
import logging from transformers import logging as tf_logging
tf_logging.set_verbosity_error()
import logging
logging.getLogger("numba").setLevel(logging.WARNING) logging.getLogger("numba").setLevel(logging.WARNING)
from transformers import ( from transformers import (

16
GPT_SoVITS/text/cleaner.py
View File

@ -1,4 +1,4 @@
from text import japanese, cleaned_text_to_sequence, english,korean,cantonese from text import cleaned_text_to_sequence
import os import os
# if os.environ.get("version","v1")=="v1": # if os.environ.get("version","v1")=="v1":
# from text import chinese # from text import chinese
@ -9,8 +9,6 @@ import os
from text import symbols as symbols_v1 from text import symbols as symbols_v1
from text import symbols2 as symbols_v2 from text import symbols2 as symbols_v2
from text import chinese as chinese_v1
from text import chinese2 as chinese_v2
special = [ special = [
# ("%", "zh", "SP"), # ("%", "zh", "SP"),
@ -24,10 +22,10 @@ def clean_text(text, language, version=None):
if version is None:version=os.environ.get('version', 'v2') if version is None:version=os.environ.get('version', 'v2')
if version == "v1": if version == "v1":
symbols = symbols_v1.symbols symbols = symbols_v1.symbols
language_module_map = {"zh": chinese_v1, "ja": japanese, "en": english} language_module_map = {"zh": "chinese", "ja": "japanese", "en": "english"}
else: else:
symbols = symbols_v2.symbols symbols = symbols_v2.symbols
language_module_map = {"zh": chinese_v2, "ja": japanese, "en": english, "ko": korean,"yue":cantonese} language_module_map = {"zh": "chinese2", "ja": "japanese", "en": "english", "ko": "korean","yue":"cantonese"}
if(language not in language_module_map): if(language not in language_module_map):
language="en" language="en"
@ -35,7 +33,7 @@ def clean_text(text, language, version=None):
for special_s, special_l, target_symbol in special: for special_s, special_l, target_symbol in special:
if special_s in text and language == special_l: if special_s in text and language == special_l:
return clean_special(text, language, special_s, target_symbol, version) return clean_special(text, language, special_s, target_symbol, version)
language_module = language_module_map[language] language_module = __import__("text."+language_module_map[language],fromlist=[language_module_map[language]])
if hasattr(language_module,"text_normalize"): if hasattr(language_module,"text_normalize"):
norm_text = language_module.text_normalize(text) norm_text = language_module.text_normalize(text)
else: else:
@ -62,16 +60,16 @@ def clean_special(text, language, special_s, target_symbol, version=None):
if version is None:version=os.environ.get('version', 'v2') if version is None:version=os.environ.get('version', 'v2')
if version == "v1": if version == "v1":
symbols = symbols_v1.symbols symbols = symbols_v1.symbols
language_module_map = {"zh": chinese_v1, "ja": japanese, "en": english} language_module_map = {"zh": "chinese", "ja": "japanese", "en": "english"}
else: else:
symbols = symbols_v2.symbols symbols = symbols_v2.symbols
language_module_map = {"zh": chinese_v2, "ja": japanese, "en": english, "ko": korean,"yue":cantonese} language_module_map = {"zh": "chinese2", "ja": "japanese", "en": "english", "ko": "korean","yue":"cantonese"}
""" """
特殊静音段sp符号处理 特殊静音段sp符号处理
""" """
text = text.replace(special_s, ",") text = text.replace(special_s, ",")
language_module = language_module_map[language] language_module = __import__("text."+language_module_map[language],fromlist=[language_module_map[language]])
norm_text = language_module.text_normalize(text) norm_text = language_module.text_normalize(text)
phones = language_module.g2p(norm_text) phones = language_module.g2p(norm_text)
new_ph = [] new_ph = []