parallelformers icon indicating copy to clipboard operation
parallelformers copied to clipboard

Published 20 hours ago • verified publisher icon tunib-ai

Can you please add support for gpt_neox

Open tahercoolguy opened this issue 2 years ago • 2 comments

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

tahercoolguy avatar Aug 11 '22 04:08 tahercoolguy

Can you make a policy of the model?

hyunwoongko avatar Aug 11 '22 12:08 hyunwoongko

You can refer to https://github.com/tunib-ai/parallelformers/blob/main/parallelformers/policies

hyunwoongko avatar Aug 11 '22 12:08 hyunwoongko