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
1da84ae02c
Fix Bug in Flax-Speech-Encoder-Decoder Test ( #16041 )
...
* Fix Bug in Flax-Speech-Encoder-Decoder Test
* change thresholds for CPU precision
2022-03-10 12:09:29 +01:00
Sanchit Gandhi
fde901877a
Freeze Feature Encoder in FlaxSpeechEncoderDecoder ( #15997 )
...
* Freeze Feature Encoder in FlaxSpeechEncoderDecoder
* add backprop test
2022-03-10 09:59:19 +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
Sanchit Gandhi
e3342edc4e
Flax Speech-Encoder-Decoder Model ( #15613 )
...
* rebase
* Delete shift tokens func
* downsample decoder input seq len for init
* correct attention mask
* add tests
* pt flax cross test
* make fixup
* init file for import
* change pt-flax cross test threshold
* pt-flax test logits only
* move tests
* make repo-consistency
* consistent indentation
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com >
2022-02-28 12:22:36 +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
