Fix

huggingface · Jun 5, 2024 · 4b6cad3 · 4b6cad3
1 parent 7cc42bf
commit 4b6cad3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/models/auto/image_processing_auto.py b/src/transformers/models/auto/image_processing_auto.py
@@ -151,6 +151,8 @@
     # If the fast image processor is not defined, or torchvision is not available, we set it to None
     if not fast_image_processor_class or fast_image_processor_class[0] is None or not is_torchvision_available():
         fast_image_processor_class = None
+    else:
+        fast_image_processor_class = fast_image_processor_class[0]
 
     IMAGE_PROCESSOR_MAPPING_NAMES[model_type] = (slow_image_processor_class, fast_image_processor_class)