Skip to content

GitLab

Unverified Commit 73a0496c authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[tests/VisionEncoderDecoder] import to_2tuple from test utils (#17865)

parent bc7a6fdc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py
View file @ 73a0496c
......@@ -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():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment