peter-sk d65b14ed67 added GPTNeoForTokenClassification (#22908) ... * added GPTNeoForTokenClassification * add to top-level init * fixup * test * more fixup * add to gpt_neo.mdx * repo consistency * dummy copy * fix copies * optax >= 0.1.5 assumes jax.Array exists - which it doesn't for jax <= 0.3.6 * merge with main made this superfluous * added classifier_dropout * remove legacy code * removed fmt:on/off removed expected_outputs * doc style fix * classifier_dropout is always in config --------- Co-authored-by: Prof. Peter Schneider-Kamp <jps@ordbogen.com>		2023-04-27 12:10:03 -04:00
..
__init__.py	Move test model folders (#17034)	2022-05-03 14:42:02 +02:00
test_modeling_flax_gpt_neo.py	Update quality tooling for formatting (#21480)	2023-02-06 18:10:56 -05:00
test_modeling_gpt_neo.py	added GPTNeoForTokenClassification (#22908)	2023-04-27 12:10:03 -04:00