Model trained on the TinyStories Dataset, see https://arxiv.org/abs/2305.07759 EXAMPLE USAGE from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig model = AutoModelForCausalLM.from pretrained('roneneldan/TinyStories 3M') tokenizer = AutoTokenizer.from pretrained("EleutherAI/gpt neo 125M") prompt = "Once upon a time there was" input ids = tokenizer.encode(prompt, return tensors="pt") Generate completion output = model.generate(input ids, max length = 1000, num beams=1) Decode the completion output text = tokenizer.decode(output[0], skip special tokens=True) Print the generated text print(output text)
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy