File size: 203 Bytes
360e70e
 
139bee4
360e70e
 
 
 
0c33b7e
 
360e70e
1
2
3
4
5
6
7
8
9
10
import transformers 
from transformers import AutoModelForCausalLM
model_id = "gpt2"

model = AutoModelForCausalLM.from_pretrained(model_id)

text = "Hello"

outputs = model.generate(text)
print(outputs)