jsnfly
51e0ebedcb
Allow passing encoder_ouputs as tuple to EncoderDecoder Models ( #16814 )
...
* Add passing encoder_outputs as tuple to existing test
* Add check for tuple
* Add check for tuple also for speech and vision
Co-authored-by: jsnfly <jsnfly@gmx.de>
2022-04-18 19:49:58 +02:00
Joao Gante
7a9ef8181c
TF: properly handle kwargs in encoder_decoder architectures ( #16465 )
...
* properly handle kwargs in encoder_decoder architectures
* make fixup
2022-03-29 18:17:47 +01:00
Sanchit Gandhi
2de99e6c43
Fix Loading of Flax(Speech)EncoderDecoderModel kwargs from PreTrained Encoder-Decoder Checkpoints ( #16056 )
...
* Fix Loading of Flax(Speech)EncoderDecoderModel kwargs from PreTrained Encoder-Decoder Checkpoints
* change wording
2022-03-14 10:12:29 +01:00
Sanchit Gandhi
b256f3518d
Add FlaxBartForCausalLM ( #15995 )
...
* add causal lm
* add CausalLM tests
* Add FlaxBartForCausalLM
* Add EncoderDecoder model tests
* change docstring
* make repo-consistency
* suggested changes
* remove jax ops
* correction
* rename pre-trained decoder model
2022-03-09 19:53:01 +01:00
Yih-Dar
b7fa1e3dee
Use tiny models for get_pretrained_model in TFEncoderDecoderModelTest ( #15989 )
...
* Use tiny model for TFRembertEncoderDecoderModelTest.get_pretrained_model()
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-03-09 17:16:25 +01:00
Yih-Dar
72983303c5
Fix TFEncoderDecoderModelTest - Pytorch device ( #15979 )
...
* fix device
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-03-08 13:37:20 +01:00
Lysandre Debut
29c10a41d0
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
...
* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>
2022-02-23 15:46:28 -05:00
