| thon | |
| from transformers import GPTNeoForCausalLM, GPT2Tokenizer | |
| model = GPTNeoForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B") | |
| tokenizer = GPT2Tokenizer.from_pretrained("EleutherAI/gpt-neo-1.3B") | |
| prompt = ( | |
| "In a shocking finding, scientists discovered a herd of unicorns living in a remote, " | |
| "previously unexplored valley, in the Andes Mountains. |