Downloads · 30 days
209K
10% of all-time downloads
TinyStories-1M
roneneldan/TinyStories-1M
TinyStories-1M is a text generation model from roneneldan. Use it when you need the model to write or continue text. It is set up for transformers.
Model trained on the TinyStories Dataset, see https://arxiv.org/abs/2305.07759
All-time downloads
2M
Public
Repo size
135 MB
Likes
74
Trending 1
Repo makeup
Click a slice to open those files.
.bin48.6 MB · 94%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt_neo
- Access
- Public
- Created
- May 12, 2023
- Updated
- Dec 18, 2025
- SHA
- 77f1b168
Try a prompt
Datasets
Model card
From the Hugging Face model README
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-1M')
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)
- Task
- Text Generation
- Library
- transformers
- Type
- gpt_neo
- Created
- May 12, 2023
- Updated
- Dec 18, 2025