j0no12 commited on
Commit
616e7df
·
verified ·
1 Parent(s): e35aa65

Rewrite model card for clarity

Browse files
Files changed (1) hide show
  1. README.md +32 -118
README.md CHANGED
@@ -16,36 +16,37 @@ tags:
16
 
17
  # Sol Lite Base
18
 
19
- Sol Lite Base is a compact **14,995,843-parameter** decoder-only language model trained from scratch on **8,153,333,760 tokens**.
20
 
21
- It combines grouped-query attention, selective recurrent depth, XSA value subtraction, loop conditioning, and a small learned n-gram memory.
22
 
23
- **This is a base model, not an instruction-tuned assistant.**
24
 
25
- ## Model summary
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
26
 
27
- | Property | Value |
28
- | --- | --- |
29
- | Parameters | **14,995,843** |
30
- | Training tokens | **8,153,333,760** |
31
- | Architecture | Decoder-only causal language model |
32
- | Physical / effective blocks | 10 / 14 |
33
- | Residual width | 256 |
34
- | Vocabulary | 4,096-token digit-aware byte-level BPE |
35
- | Maximum context | 2,048 tokens |
36
- | Weight format | FP32 safetensors |
37
- | License | CC BY 4.0 |
38
 
39
- ## Quick start
40
-
41
- Install the runtime dependencies:
42
 
43
  ```bash
44
  pip install "torch>=2.5" "transformers>=5" safetensors huggingface_hub
45
  ```
46
 
47
- Load the model and generate text:
48
-
49
  ```python
50
  from huggingface_hub import snapshot_download
51
  import sys
@@ -67,115 +68,28 @@ text = generate(
67
  print(text)
68
  ```
69
 
70
- Sol Lite Base uses the included custom PyTorch class `SolForCausalLM`.
71
-
72
- The repository includes the complete standalone model implementation and does not require the original training code.
73
-
74
- ## Results
75
 
76
- | Benchmark | Examples | Score |
77
- | --- | ---: | ---: |
78
  | HellaSwag | 10,042 | 27.72% |
79
  | ARC-Easy | 2,376 | 34.72% |
80
  | ARC-Challenge | 1,172 | 22.87% |
81
  | PIQA | 1,838 | 57.73% |
82
  | ArithMark-3 | 1,000 | 34.20% |
83
- | Intelligence Index | — | **8.799** |
84
-
85
- The four language-model tasks were evaluated zero-shot on their complete evaluation splits with `lm-eval` 0.4.12 using normalized accuracy.
86
-
87
- ArithMark-3 used its independent-tokenization normalized-accuracy protocol.
88
-
89
- Raw evaluation outputs are included under `evals/`.
90
-
91
- These measurements are not independently verified, and no leaderboard position is claimed.
92
-
93
- ## Architecture
94
-
95
- | Component | Configuration |
96
- | --- | --- |
97
- | Stored parameters | **14,995,843** |
98
- | Physical / effective blocks | 10 / 14 |
99
- | Recurrent layout | 1 prelude + 4 middle blocks × 2 passes + 5 coda |
100
- | Residual width | 256 |
101
- | Attention | 8 query heads, 2 key/value heads, head dimension 32 |
102
- | Attention modification | XSA value subtraction after causal attention |
103
- | Attention normalization | Per-head Q/K RMSNorm with RoPE |
104
- | Recurrent conditioning | Learned pass embeddings and channel-wise refresh gates |
105
- | MLP | Gated SiLU MLP, width 1,465 |
106
- | Vocabulary | 4,096-token digit-aware byte-level BPE |
107
- | Learned memory | 2,048-entry bigram/trigram EngramLite |
108
- | Embeddings | Tied input/output table |
109
- | Maximum context | 2,048 tokens |
110
-
111
- ## Training data
112
-
113
- Training used a gradual English curriculum built from:
114
-
115
- - FineWeb-Edu
116
- - DCLM
117
- - English UltraFineWeb levels 1–3
118
- - FineWeb-HQ
119
- - FineMath
120
- - FinePhrase
121
-
122
- The tokenizer and training stream were frozen before training.
123
-
124
- Public benchmark examples were not used for checkpoint selection.
125
-
126
- ## Intended use
127
-
128
- Sol Lite Base is intended for research on highly parameter-efficient language modeling, including:
129
-
130
- - extremely small language models
131
- - recurrent-depth architectures
132
- - grouped-query attention at small scale
133
- - compact learned memory
134
- - distillation experiments
135
- - efficient inference research
136
-
137
- Because this is a base model, users should expect raw next-token completion behavior rather than instruction-following or assistant-style responses.
138
-
139
- ## Limitations
140
-
141
- At approximately 15M parameters, Sol Lite Base has significant capability limitations.
142
-
143
- Outputs may be:
144
-
145
- - inconsistent
146
- - factually incorrect
147
- - repetitive
148
- - incoherent
149
- - unsafe
150
-
151
- The model is not instruction tuned and should not be treated as a conversational assistant.
152
-
153
- Do not use it for high-stakes decisions.
154
-
155
- Users are responsible for evaluating outputs and determining whether the model is suitable for their application.
156
-
157
- ## Repository files
158
 
159
- | File | Purpose |
160
- | --- | --- |
161
- | `model.safetensors` | FP32 model weights |
162
- | `modeling_sol_lite.py` | Standalone model, loader, and generator |
163
- | `config.json` | Architecture configuration |
164
- | `tokenizer.json` | Frozen tokenizer |
165
- | `tokenizer_config.json` | Tokenizer metadata |
166
- | `training_state.json` | Training and provenance summary |
167
- | `evals/` | Raw benchmark results |
168
 
169
- ## Reproducibility and evaluation notes
170
 
171
- The repository includes raw evaluation outputs under `evals/`.
172
 
173
- Reported benchmark values should be interpreted as reference measurements for this release. They have not been independently verified.
174
 
175
- No leaderboard position is claimed.
176
 
177
- ## License
178
 
179
- Sol Lite Base is released under the **CC BY 4.0** license.
180
 
181
- Dataset licenses and terms remain with their respective owners.
 
16
 
17
  # Sol Lite Base
18
 
19
+ Sol Lite Base is a 14,995,843-parameter language model trained from scratch on 8,153,333,760 tokens. It generates text completions and has not been instruction-tuned.
20
 
21
+ The model reuses four transformer blocks for a second pass. Grouped-query attention, loop conditioning, XSA value subtraction, and EngramLite local memory make up the rest of the architecture.
22
 
23
+ ## Model
24
 
25
+ | Setting | Value |
26
+ |---|---|
27
+ | Parameters | 14,995,843 |
28
+ | Training tokens | 8,153,333,760 |
29
+ | Context | 2,048 tokens |
30
+ | Tokenizer | 4,096-entry digit-aware byte-level BPE |
31
+ | Hidden width | 256 |
32
+ | Stored blocks / effective applications | 10 / 14 |
33
+ | Recurrent layout | 1 prelude, 4 middle blocks used twice, 5 coda blocks |
34
+ | Attention | 8 query heads, 2 KV heads, head dimension 32 |
35
+ | Q/K normalization | Per-head RMSNorm with RoPE |
36
+ | Recurrence | Learned pass embeddings and channel-wise refresh gates |
37
+ | FFN | Gated SiLU, width 1,465 |
38
+ | Local memory | 2,048-entry bigram/trigram EngramLite |
39
+ | Token embeddings | Tied to the output head |
40
+ | Weights | FP32 safetensors |
41
 
42
+ ## Use
 
 
 
 
 
 
 
 
 
 
43
 
44
+ The repository includes a standalone PyTorch implementation, loader, and generation helper.
 
 
45
 
46
  ```bash
47
  pip install "torch>=2.5" "transformers>=5" safetensors huggingface_hub
48
  ```
49
 
 
 
50
  ```python
51
  from huggingface_hub import snapshot_download
52
  import sys
 
68
  print(text)
69
  ```
70
 
71
+ ## Evaluation
 
 
 
 
72
 
73
+ | Benchmark | Examples | Normalized accuracy |
74
+ |---|---:|---:|
75
  | HellaSwag | 10,042 | 27.72% |
76
  | ARC-Easy | 2,376 | 34.72% |
77
  | ARC-Challenge | 1,172 | 22.87% |
78
  | PIQA | 1,838 | 57.73% |
79
  | ArithMark-3 | 1,000 | 34.20% |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
80
 
81
+ The reported Intelligence Index is 8.799. The language-model tasks use the complete zero-shot evaluation splits with `lm-eval` 0.4.12. ArithMark-3 uses independent tokenization and normalized accuracy. These are local measurements; raw outputs are in `evals/`.
 
 
 
 
 
 
 
 
82
 
83
+ ## Training
84
 
85
+ The English curriculum draws from FineWeb-Edu, DCLM, UltraFineWeb levels 1-3, FineWeb-HQ, FineMath, and FinePhrase. The tokenizer and prepared training stream were fixed before training. Public benchmark examples were not used to select the checkpoint.
86
 
87
+ ## Files
88
 
89
+ `model.safetensors` contains the weights. `modeling_sol_lite.py` defines `SolForCausalLM`, the loader, and the generator. Architecture settings are in `config.json`; tokenizer files are `tokenizer.json` and `tokenizer_config.json`. `training_state.json` records training provenance.
90
 
91
+ ## Limits and license
92
 
93
+ This is a research model for small-scale language modeling, recurrence, memory, and distillation experiments. Completions can be repetitive, incoherent, or factually wrong. It has no instruction-following training or assistant safety tuning.
94
 
95
+ The model is licensed under [CC BY 4.0](LICENSE). Upstream datasets retain their own terms.