transformers/tests/models/bart
samuelpullely 15c68c67f4
Enable decoder_attention_mask in generate function (#20726)
* Enable `decoder_attention_mask` in `generate` function

* Make style corrections

* Run `make repo-consistency`

* Add integration test
2023-01-03 09:59:08 -05:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_bart.py Enable decoder_attention_mask in generate function (#20726) 2023-01-03 09:59:08 -05:00
test_modeling_flax_bart.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_modeling_tf_bart.py Add TFBartForSequenceClassification (#20570) 2022-12-07 18:05:39 +01:00
test_tokenization_bart.py Replace as_target context managers by direct calls (#18325) 2022-07-29 08:09:09 -04:00