bacow/GPT-SoVITS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1v2 Version Switching (#1391)

Browse Source 
v1v2 Version Switching
This commit is contained in:
XXXXRT666
2024-08-06 12:00:51 +08:00
committed by GitHub
parent 4e34814c70
commit a3f5fb9614
26 changed files with 307 additions and 177 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
GPT_SoVITS/text/cleaner.py
View File

@@ -20,7 +20,8 @@ special = [
]
def clean_text(text, language, version):
def clean_text(text, language, version=None):
if version is None:version=os.environ.get('version', 'v2')
if version == "v1":
symbols = symbols_v1.symbols
language_module_map = {"zh": chinese_v1, "ja": japanese, "en": english}
@@ -57,7 +58,8 @@ def clean_text(text, language, version):
return phones, word2ph, norm_text
def clean_special(text, language, special_s, target_symbol, version):
def clean_special(text, language, special_s, target_symbol, version=None):
if version is None:version=os.environ.get('version', 'v2')
if version == "v1":
symbols = symbols_v1.symbols
language_module_map = {"zh": chinese_v1, "ja": japanese, "en": english}
@@ -82,9 +84,11 @@ def clean_special(text, language, special_s, target_symbol, version):
return new_ph, phones[1], norm_text
def text_to_sequence(text, language):
def text_to_sequence(text, language, version=None):
version = os.environ.get('version',version)
if version is None:version='v2'
phones = clean_text(text)
return cleaned_text_to_sequence(phones)
return cleaned_text_to_sequence(phones, version)
if __name__ == "__main__":