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