Safer import

huggingface · Apr 26, 2024 · 08da911 · 08da911
1 parent fc226af
commit 08da911
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/transformers/image_processing_utils.py b/src/transformers/image_processing_utils.py
@@ -26,7 +26,7 @@
 from .dynamic_module_utils import custom_object_save
 from .feature_extraction_utils import BatchFeature as BaseBatchFeature
 from .image_transforms import center_crop, normalize, rescale
-from .image_utils import ChannelDimension, PILImageResampling, is_scaled_image, to_numpy_array, valid_images
+from .image_utils import ChannelDimension, is_scaled_image, to_numpy_array, valid_images
 from .utils import (
     IMAGE_PROCESSOR_NAME,
     PushToHubMixin,
@@ -44,6 +44,8 @@
 if is_vision_available():
     from PIL import Image
 
+    from .image_utils import PILImageResampling
+
 logger = logging.get_logger(__name__)