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 |
|