tests/gptj/test_modeling_tf_gptj.py · ed2ee373d07aa8fd3f97e5f9fac9649511cf46fd · chenpangpang / transformers

Add TF implementation of GPT-J (#15623) · ed2ee373

Daniel Stancl authored Mar 25, 2022

* Initial commit

* Add TFGPTJModel

* Fix a forward pass

* Add TFGPTJCausalLM

* Add TFGPTJForSequenceClassification

* Add TFGPTJForQuestionAnswering

* Fix docs

* Deal with TF dynamic shapes

* Add Loss parents to models

* Adjust split and merge heads to handle 4 and 5-dim tensors

* Update outputs for @tooslow tests

ed2ee373

test_modeling_tf_gptj.py 20.9 KB

Replace test_modeling_tf_gptj.py