Spaces:
Sleeping
Sleeping
File size: 279 Bytes
d6bf94d 360e70e d6bf94d 04020fa d6bf94d 360e70e d6bf94d 0c33b7e 9b1d780 |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import GPT2LMHeadModel, GPT2Tokenizer
model_id = "gpt2"
model = GPT2LMHeadModel.from_pretrained(model_id)
tokenizer = GPT2Tokenizer.from_pretrained(model_id)
text = "Hello"
encoded_text = tokenizer(text, return_tensors="pt")
outputs = model(**encoded_text)
|