transformers/tests/models/videomae
Raushan Turganbay bf68dd9e6e
[tests] expand flex-attn test for vision models (#38434)
* expand the test for VLMs

* typo

* mark models `supports_flex` + expand test for additional kwargs

* flex attn for refactored vision models

* fix copies

* fix

* unskip

* style

* address comments
2025-06-03 07:40:44 +00:00
..
__init__.py Add VideoMAE (#17821) 2022-08-04 18:02:55 +02:00
test_image_processing_videomae.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_modeling_videomae.py [tests] expand flex-attn test for vision models (#38434) 2025-06-03 07:40:44 +00:00