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

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_modeling_flax_speech_encoder_decoder.py

Add FlaxBartForCausalLM (#15995)

2022-03-09 19:53:01 +01:00

test_modeling_speech_encoder_decoder.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00