transformers/tests/models/vision_encoder_decoder
Patrick von Platen 8869bf41fe
[VisionEncoderDecoder] Add gradient checkpointing (#18697)
* add first generation tutorial

* VisionEnocderDecoder gradient checkpointing

* remove generation

* add tests
2022-08-26 14:11:27 +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 Update TF(Vision)EncoderDecoderModel PT/TF equivalence tests (#18073) 2022-07-18 15:29:14 +02:00
test_modeling_vision_encoder_decoder.py [VisionEncoderDecoder] Add gradient checkpointing (#18697) 2022-08-26 14:11:27 +02:00