transformers/tests/models/opt
Thomas Wang abc400b06a
Add final_layer_norm to OPT model (#17785)
* Add final_layer_norm to OPT model

* Add JAX and TF version

* Fix Keras name

* Woops

* Allow for non breaking change

* Apply suggestions from code review

* add tests

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-06-21 20:26:36 +02:00
..
__init__.py Add OPT (#17088) 2022-05-12 12:24:35 +02:00
test_modeling_flax_opt.py Add final_layer_norm to OPT model (#17785) 2022-06-21 20:26:36 +02:00
test_modeling_opt.py Add final_layer_norm to OPT model (#17785) 2022-06-21 20:26:36 +02:00
test_modeling_tf_opt.py Add final_layer_norm to OPT model (#17785) 2022-06-21 20:26:36 +02:00