EleutherAI

/

gpt-neox-20b

like 534

Follow

EleutherAI 513

This PR contains the Flax weights of GPT-NeoX 20B
FlaxGPTNeoXForCausalLM is not yet included in transformers release. This work is progressing in https://github.com/huggingface/transformers/pull/25334

I uploaded this weights for passing tests_pr_documentation_tests
https://github.com/huggingface/transformers/pull/25334#issuecomment-1733859618