From 101535d5c723a40a7d067fa515c14f17dbe3d905 Mon Sep 17 00:00:00 2001
From: Amy Roberts <22614925+amyeroberts@users.noreply.github.com>
Date: Fri, 24 May 2024 18:20:03 +0000
Subject: [PATCH] Fix up

---
 src/transformers/models/vit/image_processing_vit_fast.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/transformers/models/vit/image_processing_vit_fast.py b/src/transformers/models/vit/image_processing_vit_fast.py
index db2fcb2f29791b..ebe5b26317a2cb 100644
--- a/src/transformers/models/vit/image_processing_vit_fast.py
+++ b/src/transformers/models/vit/image_processing_vit_fast.py
@@ -17,8 +17,8 @@
 import functools
 from typing import Any, Dict, List, Optional, Union
 
-from ...image_processing_utils import get_size_dict
 from ...image_processing_base import BatchFeature
+from ...image_processing_utils import get_size_dict
 from ...image_processing_utils_fast import BaseImageProcessorFast, SizeDict
 from ...image_transforms import FusedRescaleNormalize, NumpyToTensor, Rescale
 from ...image_utils import (
@@ -44,7 +44,7 @@
 
 
 if is_torchvision_available():
-    from torchvision.transforms import Compose, PILToTensor, Resize, Lambda, Normalize
+    from torchvision.transforms import Compose, Normalize, PILToTensor, Resize
 
 
 class ViTImageProcessorFast(BaseImageProcessorFast):