Another import fix

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