raincandy-u commited on
Commit
cffa3c7
·
verified ·
1 Parent(s): 484d213

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -19
README.md CHANGED
@@ -71,36 +71,38 @@ raw_story, story, names = decode_story(
71
  print(story)
72
  ```
73
 
74
- The display helper consistently replaces numbered character tokens with `Alex`, `Robin`, and `Casey`; it does not rewrite sentences or repair errors. `raw_story` preserves the character-token text. These markers are registered as special tokens, so `skip_special_tokens=True` would remove character references; use the helper to retain them. This example generates one story. The reported evaluation used batch size eight, which changes the sampling random-number sequence even with the same seed.
 
 
75
 
76
  ## Architecture
77
 
78
- A single physical decoder block is applied four times with shared weights. Recurrence increases computation without adding four copies of the parameters.
79
 
80
- | Setting | Value |
81
  |---|---|
82
- | Unique parameters | 19,969 |
83
- | Tokenizer | WordLevel, 378 tokens |
84
- | Hidden size / FFN intermediate size | 32 / 48 |
85
- | Physical blocks / recurrent passes | 1 / 4 |
86
- | Attention | 2 query heads, 1 KV head, head dimension 16 |
87
- | Activation / position encoding | SwiGLU / RoPE |
88
- | Normalization | RMSNorm before and after attention/FFN branches |
89
- | Value residual mixing coefficient | 0.5 |
90
- | Input/output embeddings | Tied |
91
- | Maximum sequence length | 2,048 tokens |
92
- | Weight dtype / file size | FP32 / 81,364 bytes |
93
 
94
  | Parameter group | Parameters |
95
  |---|---:|
96
- | Shared input/output embedding | 12,096 |
97
- | Attention | 3,072 |
98
- | Feed-forward network | 4,608 |
99
  | RMSNorm | 160 |
100
- | Retained exit gate, with early exit disabled | 33 |
101
  | **Total** | **19,969** |
102
 
103
- Architecture code is adapted from [ByteDance/Ouro-1.4B at revision `574fa66cb8bf5abdc979642d01cf2b79b16bfab1`](https://huggingface.co/ByteDance/Ouro-1.4B/tree/574fa66cb8bf5abdc979642d01cf2b79b16bfab1). **MacroStories was initialized randomly and did not load Ouro's pretrained weights.**
 
 
104
 
105
  ## Training data and procedure
106
 
 
71
  print(story)
72
  ```
73
 
74
+ The decoder consistently replaces numbered character tokens with `Alex`, `Robin`. `raw_story` preserves the character-token text.
75
+
76
+ These markers are registered as special tokens, so `skip_special_tokens=True` would remove character references;
77
 
78
  ## Architecture
79
 
80
+ MacroStories is a decoder-only autoregressive Transformer with **19,969 parameters**. It applies one decoder block four times, sharing the block’s weights across all four passes. The input embedding and output projection also share weights.
81
 
82
+ | Component | Configuration |
83
  |---|---|
84
+ | Vocabulary | 378 WordLevel tokens |
85
+ | Hidden dimension | 32 |
86
+ | SwiGLU intermediate dimension | 48 |
87
+ | Decoder blocks | 1, applied 4 times |
88
+ | Attention | 2 query heads and 1 shared key/value head; head dimension 16 |
89
+ | Position encoding | RoPE |
90
+ | Normalization | RMSNorm before and after each attention and feed-forward sublayer, plus a final output RMSNorm |
91
+ | Value residual | In passes 2–4, average the current value projection with the first pass’s value projection |
92
+ | Input/Output embedding | Shared weight matrix |
 
 
93
 
94
  | Parameter group | Parameters |
95
  |---|---:|
96
+ | Shared embedding / output projection | 12,096 |
97
+ | Attention projections | 3,072 |
98
+ | SwiGLU projections | 4,608 |
99
  | RMSNorm | 160 |
100
+ | Inactive early-exit gate retained in the checkpoint | 33 |
101
  | **Total** | **19,969** |
102
 
103
+ The implementation is adapted from [ByteDance/Ouro](https://huggingface.co/ByteDance/Ouro-1.4B/tree/574fa66cb8bf5abdc979642d01cf2b79b16bfab1).
104
+
105
+ MacroStories was trained from random initialization; no pretrained Ouro weights were used.
106
 
107
  ## Training data and procedure
108