File size: 564 Bytes
4c9e405 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import torch
!pip install torch
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("gpt2")
tokenizer = AutoTokenizer.from_pretrained("gpt2")
inputs = tokenizer("Hugging Face is a startup based in New York City and Paris",
return_tensors="pt")
loss = model(input_ids=inputs["input_ids"],
labels=inputs["input_ids"]).loss
loss = model(input_ids=inputs["input_ids"],
labels=inputs["input_ids"]).loss
ppl = torch.exp(loss)
print(f"Perplexity: {ppl.item(): 2f}") |