transformers/tests/models/opt
Joao Gante e15f0d73db
OPT: Fix batched generation with FLAX (#21150)
* Fix Flax OPT numerical masking

* re-enable test

* add fix to bart and reintroduce copied from in opt
2023-01-18 14:24:53 +00:00
..
__init__.py Add OPT (#17088) 2022-05-12 12:24:35 +02:00
test_modeling_flax_opt.py OPT: Fix batched generation with FLAX (#21150) 2023-01-18 14:24:53 +00:00
test_modeling_opt.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_modeling_tf_opt.py add missing require_tf for TFOPTGenerationTest (#19010) 2022-09-13 18:10:11 +02:00