Fix pil_torch_interpolation_mapping import in image_processing_detr_f…

…ast (huggingface#34375) fix pil_torch_interpolation_mapping import
byi8220 · Oct 24, 2024 · 6432ad8 · 6432ad8
1 parent dd267fc
commit 6432ad8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/models/detr/image_processing_detr_fast.py b/src/transformers/models/detr/image_processing_detr_fast.py
@@ -40,7 +40,6 @@
     get_image_type,
     infer_channel_dimension_format,
     make_list_of_images,
-    pil_torch_interpolation_mapping,
     validate_annotations,
     validate_kwargs,
 )
@@ -72,7 +71,8 @@
 if is_torchvision_available():
     from torchvision.io import read_image
 
-    from ...image_utils import pil_torch_interpolation_mapping
+    if is_vision_available():
+        from ...image_utils import pil_torch_interpolation_mapping
 
     if is_torchvision_v2_available():
         from torchvision.transforms.v2 import functional as F