README.md

---
license: cdla-sharing-1.0
language:
- en
library_name: pytorch
pipeline_tag: text-generation
datasets:
- roneneldan/TinyStories
tags:
- tinystories
- small-language-model
---

# tinyLLM 29M — TinyStories

1단계 사전학습만 거친 가중치. 대화는 못 하고 이야기를 이어쓴다.

파라미터 **29,577,728** 개. RTX 3060 Ti 한 대에서 사전학습 3.06 시간 + SFT 2.4 분.
학습 코드와 설계 근거: https://github.com/sciencemj/tinyLLM

val loss 1.3202 nats/token (perplexity 3.744, bits/char 0.4659)

## 이용 조건

**이용 제약이 없다.** TinyStories 는 CDLA-Sharing-1.0 이고 §3.5 가 명시한다 — *"This Agreement imposes no obligations or restrictions on Your Use or Publication of Results."* §1.11 에서 Results 는 데이터의 Computational Use 로 얻은 산출물이며, 조건은 데이터의 de minimis 분량 이상을 포함하지 않는 것이다. 이 모델은 train/val 격차가 0.03 이라 코퍼스를 외우고 있지 않다.

## 쓰는 법

`transformers` 를 쓰지 않는다. 이 저장소의 `modeling_tinyllm.py` 하나면 된다.

```python
import torch
from tokenizers import Tokenizer
from modeling_tinyllm import TinyLM

model = TinyLM.from_pretrained(".")
tok = Tokenizer.from_file("tokenizer.json")

ids = torch.tensor([tok.encode("Once upon a time, there was a little girl named Lily.").ids])
out = model.generate(ids, 60, temperature=0.6, top_k=20)
print(tok.decode(out[0].tolist(), skip_special_tokens=True))
```

이 가중치는 대화를 못 한다. 질문을 주면 이야기의 첫 문장으로 받아
계속 써 내려간다. 대화가 필요하면
[tinyllm-29m-chat](https://huggingface.co/sciencemj/tinyllm-29m-chat) 을 쓴다.

## 구조

```
ids (B, 512)
  → nn.Embedding(8000, 512)  +  nn.Embedding(512, 512)
  → nn.TransformerEncoder(
        nn.TransformerEncoderLayer(512, nhead=8, dim_feedforward=2048,
                                   activation="gelu", norm_first=True,
                                   batch_first=True),
        num_layers=8, norm=nn.RMSNorm(512))
  → nn.Linear(512, 8000, bias=False)     # token embedding 과 tying
```

decoder-only 를 `TransformerEncoderLayer` 로 만든다. `TransformerDecoderLayer` 는
cross-attention 용 `memory` 를 필수로 요구해서 맞지 않는다.

토크나이저는 TinyStories 와 DailyDialog 합집합에서 학습한 자체 8k byte-level BPE 다.
**같이 받은 `tokenizer.json` 을 반드시 써야 한다.** 다른 토크나이저로는 동작하지 않는다.

## 한계

**된다** — 문법, 구두점, 따옴표 대화 형식, 문단 나누기, 인물 이름 유지, 인과 연결.

**안 된다** — 턴 간 기억, 질문에 대한 직접 답변, 사실성, 문장 안 반복, 논리 일관성.
영어만 안다. 사실 정보를 얻는 용도로 쓰면 안 된다.

자세한 것은 [MODEL_CARD.md](https://github.com/sciencemj/tinyLLM/blob/main/MODEL_CARD.md).

## 인용

```bibtex
@article{eldan2023tinystories,
  title={TinyStories: How Small Can Language Models Be and Still Speak Coherent English?},
  author={Eldan, Ronen and Li, Yuanzhi},
  journal={arXiv preprint arXiv:2305.07759},
  year={2023}
}
```