File size: 165 Bytes
7cc7bb3
1c0fd7a
7cc7bb3
 
 
1
2
3
4
5
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("gpt2")

model = AutoModelForCausalLM.from_pretrained("gpt2")