riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix test
Some checks are pending
Secret Leaks / trufflehog (push) Waiting to run
Details

Browse Source
This commit is contained in:
Vasqu 2025-07-01 15:58:43 +02:00
parent 38e8de3104
commit 5120ca6c8e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/vision_text_dual_encoder/test_modeling_vision_text_dual_encoder.py
View File

@ -283,6 +283,7 @@ class DeiTRobertaModelTest(VisionTextDualEncoderMixin, unittest.TestCase):
def check_vision_text_output_attention( def check_vision_text_output_attention(
self, text_config, input_ids, attention_mask, vision_config, pixel_values=None, **kwargs self, text_config, input_ids, attention_mask, vision_config, pixel_values=None, **kwargs
): ):
text_config._attn_implementation = "eager"
vision_model, text_model = self.get_vision_text_model(vision_config, text_config) vision_model, text_model = self.get_vision_text_model(vision_config, text_config)
model = VisionTextDualEncoderModel(vision_model=vision_model, text_model=text_model) model = VisionTextDualEncoderModel(vision_model=vision_model, text_model=text_model)
model.to(torch_device) model.to(torch_device)