[SeamlessM4TTokenizer] Safe import (huggingface#28026)

safe import
staghado · Jan 15, 2024 · 9d21601 · 9d21601
1 parent abc9d03
commit 9d21601
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/transformers/models/seamless_m4t/tokenization_seamless_m4t_fast.py b/src/transformers/models/seamless_m4t/tokenization_seamless_m4t_fast.py
@@ -25,11 +25,13 @@
     TextInput,
 )
 from ...tokenization_utils_fast import PreTrainedTokenizerFast
-from ...utils import PaddingStrategy, logging
-from .tokenization_seamless_m4t import (
-    SeamlessM4TTokenizer,
-)
+from ...utils import PaddingStrategy, is_sentencepiece_available, logging
+
 
+if is_sentencepiece_available():
+    from .tokenization_seamless_m4t import SeamlessM4TTokenizer
+else:
+    SeamlessM4TTokenizer = None
 
 logger = logging.get_logger(__name__)