Any2GPT
Collection
GPT2 Models by me • 5 items • Updated • 2
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("kulia-moon/tiny2GPT")
model = AutoModelForCausalLM.from_pretrained("kulia-moon/tiny2GPT")
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="kulia-moon/tiny2GPT")