Open tahercoolguy opened 2 years ago
Can you please add support for gpt_neox
Its official documentation is here https://huggingface.co/docs/transformers/model_doc/gpt_neox
Can you make a policy of the model?
You can refer to https://github.com/tunib-ai/parallelformers/blob/main/parallelformers/policies
Describe a requested feature
Can you please add support for gpt_neox
Its official documentation is here https://huggingface.co/docs/transformers/model_doc/gpt_neox