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