tests/bart/test_modeling_flax_bart.py · b256f3518d470ba53be519992c3b9d97d174db48 · chenpangpang / transformers

Add FlaxBartForCausalLM (#15995) · b256f351

Sanchit Gandhi authored Mar 09, 2022

* 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

b256f351

test_modeling_flax_bart.py 41.1 KB

Replace test_modeling_flax_bart.py