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