Another import fix

huggingface · May 16, 2024 · ffdadd2 · ffdadd2
1 parent e4e6a58
commit ffdadd2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -1181,6 +1181,7 @@
     ]
 else:
     _import_structure["image_processing_utils_fast"] = ["BaseImageProcessorFast"]
+    _import_structure["models.vit"].append("ViTImageProcessorFast")
 
 # PyTorch-backed objects
 try:
@@ -5788,6 +5789,7 @@
         from .utils.dummy_torchvision_objects import *
     else:
         from .image_processing_utils_fast import BaseImageProcessorFast
+        from .models.vit import ViTImageProcessorFast
 
     # Modeling
     try: