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." |