transformers/tests/models/pix2struct
Jungwoo Park 44bd590a29
Pix2Struct: fix wrong broadcast axis of attention mask in visual encoder (#23976)
* fix wrong broadcast axis of attention mask in visual encoder

* fix slow tests

---------

Co-authored-by: younesbelkada <younesbelkada@gmail.com>
2023-06-05 11:47:29 -04:00
..
__init__.py Add Pix2Struct (#21400) 2023-03-22 16:53:52 +01:00
test_image_processing_pix2struct.py Update test_image_processing_pix2struct.py (#22543) 2023-04-03 15:26:35 -04:00
test_modeling_pix2struct.py Pix2Struct: fix wrong broadcast axis of attention mask in visual encoder (#23976) 2023-06-05 11:47:29 -04:00
test_processor_pix2struct.py 🚨🚨🚨 [Pix2Struct] Attempts to fix training issues 🚨🚨🚨 (#23004) 2023-04-26 18:29:25 +02:00