File size: 188 Bytes
40819c2
e0a8fce
40819c2
5f81f2a
40819c2
5f81f2a
1
2
3
4
5
6
7
from transformers import TFGPT2LMHeadModel, GPT2Tokenizer

model_name = "gpt2"
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = TFGPT2LMHeadModel.from_pretrained(model_name)