wrap transformers import with try/catch

80073936 · patil-suraj · cdf26c55 · 80073936
Commit 80073936 authored Jun 16, 2022 by patil-suraj
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

src/diffusers/pipelines/grad_tts_utils.py src/diffusers/pipelines/grad_tts_utils.py +5 -1

No files found.
--- a/src/diffusers/pipelines/grad_tts_utils.py
+++ b/src/diffusers/pipelines/grad_tts_utils.py
@@ -5,7 +5,11 @@ import os
 from shutil import copyfile

 import torch
-from transformers import PreTrainedTokenizer
+
+try:
+    from transformers import PreTrainedTokenizer
+except:
+    print("transformers is not installed")

 try:
    from unidecode import unidecode