# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Digish/distilgpt2")
model = AutoModelForCausalLM.from_pretrained("Digish/distilgpt2")Quick Links
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Digish/distilgpt2")
model = AutoModelForCausalLM.from_pretrained("Digish/distilgpt2")
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Digish/distilgpt2")