From 73a0496c2f4aeeec87b6a3fe0b1f304916ef1c08 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Fri, 24 Jun 2022 15:23:30 +0200 Subject: [PATCH] [tests/VisionEncoderDecoder] import to_2tuple from test utils (#17865) --- .../test_modeling_vision_encoder_decoder.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py b/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py index f8ac8f1cdf..320cdd6330 100644 --- a/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py +++ b/tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py @@ -20,7 +20,7 @@ import unittest from datasets import load_dataset from packaging import version -from transformers.testing_utils import require_torch, require_vision, slow, torch_device +from transformers.testing_utils import require_torch, require_vision, slow, to_2tuple, torch_device from transformers.utils import cached_property, is_torch_available, is_vision_available from ...test_modeling_common import floats_tensor, ids_tensor, random_attention_mask @@ -48,7 +48,6 @@ if is_torch_available(): ViTModel, ) from transformers.modeling_outputs import BaseModelOutput - from transformers.models.vit.modeling_vit import to_2tuple if is_vision_available():