Skip to content

Commit

Permalink
Fix pil_torch_interpolation_mapping import in image_processing_detr_f…
Browse files Browse the repository at this point in the history
…ast (huggingface#34375)

fix pil_torch_interpolation_mapping import
  • Loading branch information
yonigozlan authored Oct 24, 2024
1 parent dd267fc commit 6432ad8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/transformers/models/detr/image_processing_detr_fast.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@
get_image_type,
infer_channel_dimension_format,
make_list_of_images,
pil_torch_interpolation_mapping,
validate_annotations,
validate_kwargs,
)
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 6432ad8

Please sign in to comment.