| thon | |
| from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast | |
| model = GPTNeoXForCausalLM.from_pretrained("EleutherAI/gpt-neox-20b") | |
| tokenizer = GPTNeoXTokenizerFast.from_pretrained("EleutherAI/gpt-neox-20b") | |
| prompt = "GPTNeoX20B is a 20B-parameter autoregressive Transformer model developed by EleutherAI." |