from transformers import AutoModelForCausalLM, AutoTokenizer

# Load GPT-2 small
model = AutoModelForCausalLM.from_pretrained("gpt2")
tokenizer = AutoTokenizer.from_pretrained("gpt2")