| 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) |