File size: 183 Bytes
805db4c |
1 2 3 4 5 |
from transformers import AutoModelForCausalLM, AutoTokenizer
# Load GPT-2 small
model = AutoModelForCausalLM.from_pretrained("gpt2")
tokenizer = AutoTokenizer.from_pretrained("gpt2") |