transformers/tests/models/got_ocr2
Raushan Turganbay 0c78ef6cd3
🔴 VLM: compile compatibility (#35724)
* llavas

* add mroe models

* fix `compile_forward` test for all models

* fix copies

* make style

* also doesn't support cache class

* fix some tests

* not copied from

* ci green?

* fix tests

* fix copies

* fix tests

* check with `numel` and remove `item`

* fix copies

* fix copies

* Update src/transformers/models/cohere2/modeling_cohere2.py

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

* opt remove cross attn

* gemma2

* fixup

* fixup

* fix newly added test

* maybe fixed?

* green please?

---------

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
2025-02-14 15:23:49 +01:00
..
__init__.py Add GOT-OCR 2.0 to Transformers (#34721) 2025-01-31 11:28:13 -05:00
test_image_processing_got_ocr2.py Add GOT-OCR 2.0 to Transformers (#34721) 2025-01-31 11:28:13 -05:00
test_modeling_got_ocr2.py 🔴 VLM: compile compatibility (#35724) 2025-02-14 15:23:49 +01:00
test_processor_got_ocr2.py Add GOT-OCR 2.0 to Transformers (#34721) 2025-01-31 11:28:13 -05:00