From 6432ad8bb5dec9c7ece1041767c9e208ff6b4cbb Mon Sep 17 00:00:00 2001
From: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com>
Date: Thu, 24 Oct 2024 09:22:50 -0400
Subject: [PATCH] Fix pil_torch_interpolation_mapping import in
 image_processing_detr_fast (#34375)

fix pil_torch_interpolation_mapping import
---
 src/transformers/models/detr/image_processing_detr_fast.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/transformers/models/detr/image_processing_detr_fast.py b/src/transformers/models/detr/image_processing_detr_fast.py
index 97940ab3132dda..0fa1d0ffd9dba9 100644
--- 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