transformers/tests/models/vit_hybrid
Yih-Dar 6ea3ee3cd2
Fix test_model_parallelism (#25359)
* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-08-08 10:48:45 +02:00
..
__init__.py Add BiT + ViT hybrid (#20550) 2022-12-07 11:03:39 +01:00
test_modeling_vit_hybrid.py Fix test_model_parallelism (#25359) 2023-08-08 10:48:45 +02:00