VALL-E-X / macros.py
Plachta's picture
Upload 67 files
9d434bb
raw
history blame contribute delete
493 Bytes
NUM_LAYERS = 12
NUM_HEAD = 16
N_DIM = 1024
PREFIX_MODE = 1
NUM_QUANTIZERS = 8
SAMPLE_RATE = 24000
lang2token = {
'zh': "[ZH]",
'ja': "[JA]",
"en": "[EN]",
'mix': "",
}
lang2code = {
'zh': 0,
'ja': 1,
"en": 2,
}
token2lang = {
'[ZH]': "zh",
'[JA]': "ja",
"[EN]": "en",
"": "mix"
}
code2lang = {
0: 'zh',
1: 'ja',
2: "en",
}
langdropdown2token = {
'English': "[EN]",
'中文': "[ZH]",
'日本語': "[JA]",
'Mix': "",
}