transformers/tests/models/janus
Rémi Ouazan 037acf1d10
[janus] Fix failing tests on mi3XX (#38426)
* Fix multiple devices error on Janus

* Fix AttributeError on Janus BOI token

* Initialize lm first in Janus to get correct device map

* Added expectations for Janus test_model_generate_images

* Fixed JanusVisionEncoderLayer being split across devices

* Code formatting

* Adding modeling file

* Reverted changes out of scope for this PR
2025-06-04 09:38:10 +02:00
..
__init__.py Add Janus model (#36053) 2025-04-17 09:18:51 +02:00
test_image_processing_janus.py Add Janus model (#36053) 2025-04-17 09:18:51 +02:00
test_modeling_janus.py [janus] Fix failing tests on mi3XX (#38426) 2025-06-04 09:38:10 +02:00
test_processor_janus.py [Tests] Cleanup Janus Testcase (#38311) 2025-05-23 09:29:16 +02:00