test.py

from transformers import GPT2LMHeadModel, GPT2Tokenizer
import torch

# Choose the model size: 'gpt2', 'gpt2-medium', 'gpt2-large', etc.
model_name = 'gpt2'
model = GPT2LMHeadModel.from_pretrained(model_name)
tokenizer = GPT2Tokenizer.from_pretrained(model_name)

print(model)
print(tokenizer)