transformers/tests/models/eomt
Yaswanth Gali b61023a1b7
🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122)
* Make EoMT compatible with pipeline

* Implicit patch offsets

* remove patch offsets from arg

* Modify tests

* Update example

* fix proc testcase

* Add few more args

* add pipeline test suite

* fix

* docstring fixes

* add pipeline test

* changes w.r.t review

* 🙈 MB

* should fix device mismatch

* debug

* Fixes device mismatch

* use decorator

* we can split mlp

* expected values update

---------

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
2025-07-02 12:25:26 +01:00
..
__init__.py Add EoMT Model || 🚨 Fix Mask2Former loss calculation (#37610) 2025-06-27 14:18:18 +02:00
test_image_processing_eomt.py 🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122) 2025-07-02 12:25:26 +01:00
test_modeling_eomt.py 🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122) 2025-07-02 12:25:26 +01:00