tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor: minor adjustments

Browse Source
This commit is contained in:
tsukumi
2024-03-07 03:54:02 +00:00
parent 62919e904e
commit 4f11b011fd
8 changed files with 18 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
style_bert_vits2/text_processing/symbols.py
View File

@@ -77,8 +77,8 @@ ZH_SYMBOLS = [
]
NUM_ZH_TONES = 6
# japanese
JA_SYMBOLS = [
# Japanese
JP_SYMBOLS = [
"N",
"a",
"a:",
@@ -122,7 +122,7 @@ JA_SYMBOLS = [
"z",
"zy",
]
NUM_JA_TONES = 2
NUM_JP_TONES = 2
# English
EN_SYMBOLS = [
@@ -169,23 +169,25 @@ EN_SYMBOLS = [
NUM_EN_TONES = 4
# Combine all symbols
NORMAL_SYMBOLS = sorted(set(ZH_SYMBOLS + JA_SYMBOLS + EN_SYMBOLS))
NORMAL_SYMBOLS = sorted(set(ZH_SYMBOLS + JP_SYMBOLS + EN_SYMBOLS))
SYMBOLS = [PAD] + NORMAL_SYMBOLS + PUNCTUATION_SYMBOLS
SIL_PHONEMES_IDS = [SYMBOLS.index(i) for i in PUNCTUATION_SYMBOLS]
# Combine all tones
NUM_TONES = NUM_ZH_TONES + NUM_JA_TONES + NUM_EN_TONES
NUM_TONES = NUM_ZH_TONES + NUM_JP_TONES + NUM_EN_TONES
# Language maps
LANGUAGE_ID_MAP = {"ZH": 0, "JP": 1, "EN": 2}
NUM_LANGUAGES = len(LANGUAGE_ID_MAP.keys())
# Language tone start map
LANGUAGE_TONE_START_MAP = {
"ZH": 0,
"JP": NUM_ZH_TONES,
"EN": NUM_ZH_TONES + NUM_JA_TONES,
"EN": NUM_ZH_TONES + NUM_JP_TONES,
}
if __name__ == "__main__":
a = set(ZH_SYMBOLS)
b = set(EN_SYMBOLS)