File size: 180 Bytes
360e70e
04020fa
139bee4
360e70e
04020fa
360e70e
 
0c33b7e
9b1d780
1
2
3
4
5
6
7
8
9
10
import transformers 
from transformers import GPT2LMHeadModel
model_id = "gpt2"

model = GPT2LMHeadModel.from_pretrained(model_id)

text = "Hello"

outputs = model(**encoded_text)