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 ...@@ -20,7 +20,7 @@ import unittest
from datasets import load_dataset from datasets import load_dataset
from packaging import version 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 transformers.utils import cached_property, is_torch_available, is_vision_available
from ...test_modeling_common import floats_tensor, ids_tensor, random_attention_mask from ...test_modeling_common import floats_tensor, ids_tensor, random_attention_mask
...@@ -48,7 +48,6 @@ if is_torch_available(): ...@@ -48,7 +48,6 @@ if is_torch_available():
ViTModel, ViTModel,
) )
from transformers.modeling_outputs import BaseModelOutput from transformers.modeling_outputs import BaseModelOutput
from transformers.models.vit.modeling_vit import to_2tuple
if is_vision_available(): 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