wrap inflect in try catch

2022-06-17 13:48:51 +02:00 · 2022-06-17 13:48:51 +02:00 · 800b27703e
parent d76bc43720
commit 800b27703e
1 changed files with 5 additions and 1 deletions
--- a/src/diffusers/pipelines/grad_tts_utils.py
+++ b/src/diffusers/pipelines/grad_tts_utils.py
@ -236,8 +236,12 @@ def english_cleaners(text):
    text = collapse_whitespace(text)
    return text

+try:
+    _inflect = inflect.engine()
+except:
+    print("inflect is not installed")
+    _inflect = None

-_inflect = inflect.engine()
 _comma_number_re = re.compile(r"([0-9][0-9\,]+[0-9])")
 _decimal_number_re = re.compile(r"([0-9]+\.[0-9]+)")
 _pounds_re = re.compile(r"£([0-9\,]*[0-9]+)")