Hugging Face's logo

EleutherAI
/
gpt-neox-20b

Text Generation Transformers PyTorch Safetensors
English gpt_neox causal-lm Inference Endpoints text-generation-inference 4 papers
Model card Files Community
25

Upload FlaxGPTNeoXForCausalLM

#24
by heegyu - opened
base: refs/heads/main
←
from: refs/pr/24
Discussion Files changed
+567
-1
heegyu
No description provided.
Upload FlaxGPTNeoXForCausalLMf0c78df4
heegyu

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

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment