transformers/tests/models/vision_encoder_decoder
Yih-Dar 84f6bee5da
PT <-> TF for composite models (#19732)
* First step of PT->TF for composite models

* Update the tests

* For VisionEncoderDecoderModel

* Fix

* Fix

* Add comment

* Fix

* clean up import

* Save memory

* For (TF)EncoderDecoderModel

* For (TF)EncoderDecoderModel

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-10-21 12:40:39 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_vision_encoder_decoder.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_tf_vision_encoder_decoder.py PT <-> TF for composite models (#19732) 2022-10-21 12:40:39 +02:00
test_modeling_vision_encoder_decoder.py send model to the correct device (#18800) 2022-08-29 18:46:30 +02:00