File size: 165 Bytes
6c678fc
 
 
 
 
1
2
3
4
5
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("gpt2")

model = AutoModelForCausalLM.from_pretrained("gpt2")