bug fix and check cleaner config field by comparing with the list of

avail functions
This commit is contained in:
erogol 2020-03-17 15:05:13 +01:00
Родитель fa795347a9
Коммит b9df54adcd
2 изменённых файлов: 4 добавлений и 2 удалений

Просмотреть файл

@ -496,7 +496,8 @@ def check_config(c):
_check_argument('tb_model_param_stats', c, restricted=True, val_type=bool)
# dataloading
_check_argument('text_cleaner', c, restricted=True, val_type=str, enum_list=['english_cleaners', 'phoneme_cleaners', 'transliteration_cleaners', 'basic_cleaners'])
from TTS.utils.text import cleaners
_check_argument('text_cleaner', c, restricted=True, val_type=str, enum_list=dir(cleaners))
_check_argument('enable_eos_bos_chars', c, restricted=True, val_type=bool)
_check_argument('num_loader_workers', c, restricted=True, val_type=int, min_val=0)
_check_argument('num_val_loader_workers', c, restricted=True, val_type=int, min_val=0)

Просмотреть файл

@ -97,6 +97,7 @@ def basic_turkish_cleaners(text):
text = text.replace("I", "ı")
text = lowercase(text)
text = collapse_whitespace(text)
return text
def english_cleaners(text):