# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Gaphilly/gpt2_shakespeare")
model = AutoModelForCausalLM.from_pretrained("Gaphilly/gpt2_shakespeare", device_map="auto")Quick Links
I fine-tuned gpt2-small with a collection of Shakespeare's texts: https://www.gutenberg.org/cache/epub/100/pg100.txt This literally took like 24 hours non-stop on my potato computer lol
- Downloads last month
- 5
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Gaphilly/gpt2_shakespeare")