CoolFace
Modelpublic

plimb/gladios-tiny.story-0.1B

sourceHugging Facemitupdated 2h agoView on Hugging Face
2likes204downloads
Model Card

<p align="center"> <img src="logo.png" alt="Gladios Tiny Story logo" width="200"/> </p>

<h1 align="center">gladios-tiny.story-0.1B</h1>

<p align="center"> <a href="https://huggingface.co/spaces/rusher-code/gladios-tiny-story-demo/"><img alt="Demo" src="https://img.shields.io/badge/%F0%9F%A4%97%20Demo-Gradio%20%2F%20ZeroGPU-blue"></a> <a href="https://github.com/plimb-ai/gladios-tiny-story"><img alt="GitHub" src="https://img.shields.io/badge/GitHub-code-black"></a> </p>

GGUF

A GGUF version (for llama.cpp, Ollama, LM Studio) is available thanks to mradermacher: mradermacher/gladios-tiny.story-0.1B-GGUF

๐Ÿง  Overview

gladios-tiny.story-0.1B is a decoder-only Transformer (GPT-2-style architecture, ~124M / 0.1B parameters) trained from scratch on the full TinyStories dataset (~2.1 million children's stories, ~470M tokens).

The model uses the GPT-2 tokenizer (50,257-token vocabulary) and generates short, coherent stories in English, in the style of the training data.

๐Ÿ‘‰ Try it without installing anything: Gradio demo (ZeroGPU) ๐Ÿ‘‰ Full training code: github.com/plimb-ai/gladios-tiny-story

๐Ÿ—๏ธ Architecture

TypeDecoder-only Transformer (GPT-2 architecture)
Parameters~124M (0.1B)
Layers12
Attention heads12
Embedding dim768
Context length512 tokens
TokenizerGPT-2 (tiktoken / 50,257-token vocab)
Weights`safetensors`

๐Ÿš€ Usage

python
from transformers import pipeline

gen = pipeline("text-generation", model="plimb/gladios-tiny.story-0.1B")
print(gen(
    "Once upon a time",
    max_new_tokens=200,
    do_sample=True,
    temperature=0.8,
    top_k=50,
)[0]["generated_text"])

Or directly with AutoModelForCausalLM:

python
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model = AutoModelForCausalLM.from_pretrained("plimb/gladios-tiny.story-0.1B")
tok = AutoTokenizer.from_pretrained("plimb/gladios-tiny.story-0.1B")

ids = tok("Once upon a time", return_tensors="pt").input_ids
out = model.generate(ids, max_new_tokens=200, do_sample=True, temperature=0.8, top_k=50,
                      pad_token_id=tok.eos_token_id)
print(tok.decode(out[0], skip_special_tokens=True))

๐Ÿ“š Training data

  • โ€”Dataset: roneneldan/TinyStories (full train split)
  • โ€”Tokenizer: GPT-2, with each story separated by the special <|endoftext|> token
  • โ€”Trained on the entire dataset (no subsampling)

โš ๏ธ Limitations

  • โ€”Only writes short children's stories, in English
  • โ€”No general world knowledge, no instruction-following, no conversational ability
  • โ€”May hallucinate or lose coherence on prompts far outside the TinyStories style

๐Ÿ“„ License

MIT