diff --git a/TTS/bin/synthesize.py b/TTS/bin/synthesize.py index 5ded3067..6adb9f03 100755 --- a/TTS/bin/synthesize.py +++ b/TTS/bin/synthesize.py @@ -392,7 +392,7 @@ If you don't specify any models, then it uses LJSpeech based English model. if args.encoder_path is not None: encoder_path = args.encoder_path encoder_config_path = args.encoder_config_path - + device = args.device if args.use_cuda: device = "cuda" diff --git a/TTS/tts/utils/text/belarusian/phonemizer.py b/TTS/tts/utils/text/belarusian/phonemizer.py index 3c07a209..1922577e 100644 --- a/TTS/tts/utils/text/belarusian/phonemizer.py +++ b/TTS/tts/utils/text/belarusian/phonemizer.py @@ -8,7 +8,9 @@ def init(): import jpype import jpype.imports except ModuleNotFoundError: - raise ModuleNotFoundError("Belarusian phonemizer requires to install module 'jpype1' manually. Try `pip install jpype1`.") + raise ModuleNotFoundError( + "Belarusian phonemizer requires to install module 'jpype1' manually. Try `pip install jpype1`." + ) try: jar_path = os.environ["BEL_FANETYKA_JAR"] @@ -31,4 +33,5 @@ def belarusian_text_to_phonemes(text: str) -> str: init() from org.alex73.fanetyka.impl import FanetykaText + return str(FanetykaText(finder, text).ipa) diff --git a/TTS/tts/utils/text/phonemizers/__init__.py b/TTS/tts/utils/text/phonemizers/__init__.py index 638184fd..f9a0340c 100644 --- a/TTS/tts/utils/text/phonemizers/__init__.py +++ b/TTS/tts/utils/text/phonemizers/__init__.py @@ -1,6 +1,6 @@ from TTS.tts.utils.text.phonemizers.bangla_phonemizer import BN_Phonemizer -from TTS.tts.utils.text.phonemizers.belarusian_phonemizer import BEL_Phonemizer from TTS.tts.utils.text.phonemizers.base import BasePhonemizer +from TTS.tts.utils.text.phonemizers.belarusian_phonemizer import BEL_Phonemizer from TTS.tts.utils.text.phonemizers.espeak_wrapper import ESpeak from TTS.tts.utils.text.phonemizers.gruut_wrapper import Gruut from TTS.tts.utils.text.phonemizers.ko_kr_phonemizer import KO_KR_Phonemizer diff --git a/TTS/tts/utils/text/phonemizers/belarusian_phonemizer.py b/TTS/tts/utils/text/phonemizers/belarusian_phonemizer.py index fb620766..e5fcab6e 100644 --- a/TTS/tts/utils/text/phonemizers/belarusian_phonemizer.py +++ b/TTS/tts/utils/text/phonemizers/belarusian_phonemizer.py @@ -1,7 +1,7 @@ from typing import Dict -from TTS.tts.utils.text.phonemizers.base import BasePhonemizer from TTS.tts.utils.text.belarusian.phonemizer import belarusian_text_to_phonemes +from TTS.tts.utils.text.phonemizers.base import BasePhonemizer _DEF_BE_PUNCS = ",!." # TODO diff --git a/recipes/bel-alex73/train_glowtts.py b/recipes/bel-alex73/train_glowtts.py index 24b62d79..74866be7 100644 --- a/recipes/bel-alex73/train_glowtts.py +++ b/recipes/bel-alex73/train_glowtts.py @@ -60,7 +60,7 @@ config = GlowTTSConfig( output_path=output_path, add_blank=True, datasets=[dataset_config], -# characters=characters, + # characters=characters, enable_eos_bos_chars=True, mixed_precision=False, save_step=10000, diff --git a/tests/text_tests/test_belarusian_phonemizer.py b/tests/text_tests/test_belarusian_phonemizer.py index 278ee8be..76ba4667 100644 --- a/tests/text_tests/test_belarusian_phonemizer.py +++ b/tests/text_tests/test_belarusian_phonemizer.py @@ -1,6 +1,6 @@ import os -import warnings import unittest +import warnings from TTS.tts.utils.text.belarusian.phonemizer import belarusian_text_to_phonemes @@ -17,7 +17,8 @@ class TestText(unittest.TestCase): except KeyError: warnings.warn( "You need to define 'BEL_FANETYKA_JAR' environment variable as path to the fanetyka.jar file to test Belarusian phonemizer", - Warning) + Warning, + ) return for line in _TEST_CASES.strip().split("\n"):