SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Suraj Patil
2022-06-24 15:23:30 +02:00
committed by GitHub
parent bc7a6fdc02
commit 73a0496c2f
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/models/vision_encoder_decoder/test_modeling_vision_encoder_decoder.py
View File

@@ -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():
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():