From c81f8d323cddd398ead5c70986e6a5d3080f12f2 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 5 Jul 2022 02:49:27 -0400 Subject: [PATCH] fix cmudict initialization --- tts/vocodes_common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tts/vocodes_common.py b/tts/vocodes_common.py index 1753577..edb2a96 100644 --- a/tts/vocodes_common.py +++ b/tts/vocodes_common.py @@ -32,7 +32,7 @@ import os from text.pipeline.english_pipeline_v1 import EnglishPipelineV1 from text.pipeline.legacy_fakeyou_pipeline import LegacyFakeYouPipeline -from text.pipeline_components.phonetization.cmudict import DEFAULT_CMUDICT_PATH +from text.pipeline_components.phonetization.cmudict import DEFAULT_CMUDICT_PATH, CMUDict from text.pipeline_components.phonetization.english_phonetization import EnglishPhonetization def print_gpu_info(): @@ -88,7 +88,7 @@ def load_hifigan_model(checkpoint_path, super_resolution = False): denoiser = HifiDenoiser(hifigan, mode="normal") return hifigan, h, denoiser -ENGLISH_PIPELINE_V1 = EnglishPipelineV1(EnglishPhonetization(DEFAULT_CMUDICT_PATH)) +ENGLISH_PIPELINE_V1 = EnglishPipelineV1(EnglishPhonetization(CMUDict(DEFAULT_CMUDICT_PATH))) LEGACY_FAKEYOU_PIPELINE = LegacyFakeYouPipeline() def preprocess_text(raw_user_text: str):