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