From e97126314cc39c5b346670fe2489ca463c0abba7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 21 Apr 2021 13:12:35 +0200 Subject: [PATCH] add ```unique``` argument to make_symbols to fix the incompat. issue of the SC-Glow models --- TTS/tts/utils/text/symbols.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/TTS/tts/utils/text/symbols.py b/TTS/tts/utils/text/symbols.py index 8b9c683e..d8e9a87f 100644 --- a/TTS/tts/utils/text/symbols.py +++ b/TTS/tts/utils/text/symbols.py @@ -8,9 +8,15 @@ through Unidecode. For other data, you can modify _characters. See TRAINING_DATA def make_symbols( - characters, phonemes=None, punctuations="!'(),-.:;? ", pad="_", eos="~", bos="^", unique=True, + characters, + phonemes=None, + punctuations="!'(),-.:;? ", + pad="_", + eos="~", + bos="^", + unique=True, ): # pylint: disable=redefined-outer-name - """ Function to create symbols and phonemes + """Function to create symbols and phonemes TODO: create phonemes_to_id and symbols_to_id dicts here.""" _symbols = list(characters) _symbols = [bos] + _symbols if len(bos) > 0 and bos is not None else _symbols @@ -18,7 +24,9 @@ def make_symbols( _symbols = [pad] + _symbols if len(bos) > 0 and pad is not None else _symbols _phonemes = None if phonemes is not None: - _phonemes_sorted = sorted(list(set(phonemes))) if unique else sorted(list(phonemes)) # this is to keep previous models compatible. + _phonemes_sorted = ( + sorted(list(set(phonemes))) if unique else sorted(list(phonemes)) + ) # this is to keep previous models compatible. # Prepend "@" to ARPAbet symbols to ensure uniqueness (some are the same as uppercase letters): _arpabet = ["@" + s for s in _phonemes_sorted] # Export all symbols: