Views
No views yet
A from-scratch, CPU-trained storyteller transformer. ~863,492 parameters.
transformers — every layer is implemented by hand in plain PyTorch. token + position embeddings
|
+----------------+
| Transformer x4 |
| causal MHA |
| GELU FFN |
+----------------+
|
LayerNorm
|
tied output head| hyperparameter | value |
|---|---|
| vocab size | 10000 |
| d_model | 128 |
| layers | 4 |
| heads | 4 |
| ffn dim | 512 |
| max seq len | 64 |
| parameters | 863,492 |
| tied embeddings | True |

1# files needed next to this script:
2# model.safetensors, config.json, tokenizer.json, modeling_aetherstory.py
3from modeling_aetherstory import StoryTeller
4
5teller = StoryTeller.from_dir(".")
6print(teller("In the Glasslands there lived", max_tokens=100, temperature=0.9))