transformers/tests/bart
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_bart.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_flax_bart.py Add FlaxBartForCausalLM (#15995) 2022-03-09 19:53:01 +01:00
test_modeling_tf_bart.py TF generate refactor - past without encoder outputs (#15944) 2022-03-08 14:46:44 +00:00
test_tokenization_bart.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00