protect imports in type checking

huggingface · Oct 18, 2023 · 2ae2dad · 2ae2dad
1 parent 8de6382
commit 2ae2dad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/fuyu/__init__.py b/src/transformers/models/fuyu/__init__.py
@@ -47,7 +47,7 @@
     from .configuration_fuyu import FUYU_PRETRAINED_CONFIG_ARCHIVE_MAP, FuyuConfig
 
     try:
-        if not is_vision_available():
+        if not is_vision_available() or not is_torch_available():
             raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
         pass