E6E831728 commited on
Commit
458999f
·
verified ·
1 Parent(s): 2791ac5

Delete .ipynb_checkpoints/README-checkpoint.md

Browse files
.ipynb_checkpoints/README-checkpoint.md DELETED
@@ -1,194 +0,0 @@
1
- ---
2
- library_name: transformers
3
- pipeline_tag: text-generation
4
- license: apache-2.0
5
- language:
6
- - en
7
- tags:
8
- - causal-lm
9
- - base-model
10
- - custom-code
11
- - safetensors
12
- - frozen-embeddings
13
- ---
14
-
15
- # AB-EXT 1.7B Binary16
16
-
17
- A decoder-only causal language model whose trainable input embedding table is replaced by canonical fixed 16-bit token-ID codes, repeated deterministically to the model width. The output head remains trainable and untied.
18
-
19
- This is a **base pretrained causal language model**. It has not been
20
- instruction-tuned, preference-optimized, or aligned as a conversational
21
- assistant.
22
-
23
- ## Model summary
24
-
25
- | Property | Value |
26
- |---|---|
27
- | Architecture | Decoder-only causal Transformer |
28
- | Hidden width | 2048 |
29
- | Layers | 24 |
30
- | Attention heads | 32 |
31
- | FFN intermediate width | 8192 |
32
- | Context used in training | 2048 tokens |
33
- | Position encoding | RoPE |
34
- | Activation | SwiGLU |
35
- | Normalization | RMSNorm |
36
- | Input interface | Canonical fixed 16-bit input codes |
37
- | Output head | Trainable, untied |
38
- | Training objective | Causal next-token prediction |
39
- | Training tokens | 100,000,071,680 prediction targets |
40
- | Tokenizer | `HuggingFaceTB/SmolLM2-1.7B` |
41
- | Tokenizer revision | `effd688a12921b4cc83e3312b6feb579f70f9c71` |
42
-
43
- ## Input representation
44
-
45
-
46
- For token ID $t$, the model constructs its canonical little-endian binary code:
47
-
48
- $$
49
- c(t) \in \{0,1\}^{16}.
50
- $$
51
-
52
- Because $d=2048$, the code is repeated $128$ times:
53
-
54
- $$
55
- x(t) =
56
- \underbrace{
57
- c(t) \Vert c(t) \Vert \cdots \Vert c(t)
58
- }_{128\text{ copies}}
59
- \in \mathbb{R}^{2048}.
60
- $$
61
-
62
- There is no trainable input embedding parameter or trainable input
63
- projection. The implementation stores the procedurally generated codebook
64
- as a persistent non-trainable buffer. The output vocabulary projection is
65
- trainable and untied.
66
-
67
-
68
- ## Architecture
69
-
70
- The contextual backbone uses causal multi-head self-attention, RoPE,
71
- RMSNorm and SwiGLU residual blocks. The experiment changes the input
72
- interface while retaining the same contextual backbone and tokenizer.
73
-
74
- The model is not initialized from SmolLM2 weights. SmolLM2 supplies only
75
- the tokenizer artifacts.
76
-
77
- ## Training
78
-
79
- - **Data:** FineWeb-Edu
80
- - **Validation:** FineWeb-Edu
81
- - **Objective:** causal next-token prediction
82
- - **Optimizer:** AdamW
83
- - **Peak learning rate:** `0.00015`
84
- - **Minimum learning rate:** `1e-05`
85
- - **Warmup:** `2000` optimizer steps
86
- - **Weight decay:** `0.01`
87
- - **Sequence length:** `2048`
88
- - **Precision:** FP32 master parameters and AdamW states; BF16 autocast during forward/backward
89
- - **Hardware:** 2 x NVIDIA H100
90
- - **Seed:** `42`
91
-
92
- The stated token count is the trainer's actual prediction-target counter,
93
- not merely the requested stopping threshold.
94
-
95
-
96
- ## Evaluation
97
-
98
- A frozen evaluation table will be added after all released models and
99
- comparison baselines have been evaluated with the same pinned commit of
100
- `lm-evaluation-harness`.
101
-
102
- | Benchmark | Shots | Score |
103
- |---|---:|---:|
104
- | HellaSwag | 0 | Pending |
105
- | ARC-Easy | 0 | Pending |
106
- | ARC-Challenge | 0 | Pending |
107
- | PIQA | 0 | Pending |
108
- | WinoGrande | 0 | Pending |
109
- | OpenBookQA | 0 | Pending |
110
- | CommonsenseQA | 0 | Pending |
111
- | MMLU | 5 | Pending |
112
- | TriviaQA | 0 | Pending |
113
- | GSM8K | 5 | Pending |
114
-
115
- The exact harness commit, task versions, prompts, random seeds and raw JSON
116
- outputs will be published with the scores.
117
-
118
-
119
- ## Usage
120
-
121
- This repository contains custom Transformers code. Review the files before
122
- enabling remote code execution.
123
-
124
- ```python
125
- import torch
126
- from transformers import AutoModelForCausalLM, AutoTokenizer
127
-
128
- model_id = "E6E831728/AB-EXT-1.7B-Binary16"
129
-
130
- tokenizer = AutoTokenizer.from_pretrained(model_id)
131
-
132
- model = AutoModelForCausalLM.from_pretrained(
133
- model_id,
134
- trust_remote_code=True,
135
- torch_dtype=torch.bfloat16,
136
- device_map="auto",
137
- ).eval()
138
-
139
- inputs = tokenizer(
140
- "Gravity is",
141
- return_tensors="pt",
142
- add_special_tokens=False,
143
- ).to(model.device)
144
-
145
- with torch.no_grad():
146
- output = model.generate(
147
- **inputs,
148
- max_new_tokens=32,
149
- do_sample=False,
150
- )
151
-
152
- print(tokenizer.decode(output[0], skip_special_tokens=True))
153
- ```
154
-
155
- For strict reproducibility, pin a Hub commit through `revision=...`.
156
-
157
- ## Limitations
158
-
159
- - This is a base completion model, not a chat assistant.
160
- - The training context was 2048 tokens.
161
- - Factual outputs may be incorrect or outdated.
162
- - Generated text can contain biases or harmful content inherited from the
163
- pretraining data.
164
- - Benchmark scores are sensitive to task version, prompt construction,
165
- few-shot protocol and decoding settings.
166
- - Comparisons with SmolLM2 are not controlled comparisons of training
167
- compute or data: SmolLM2 uses a different architecture, corpus and token
168
- budget.
169
- - The tokenizer is reused from SmolLM2, but the model weights are newly
170
- trained and are not derivatives of SmolLM2 weights.
171
-
172
- ## Reproducibility and integrity
173
-
174
- - Weight SHA-256: `c3e78f24f03dff36b6174985dc5189e8ce37a52685d022ade36fa68c3239801b`
175
- - Stored tensor values: `1,712,162,816`
176
- - Stored values by dtype: `{"F32": 1712162816}`
177
- - Custom loading requires `trust_remote_code=True`.
178
-
179
- ## Attribution
180
-
181
- The tokenizer artifacts are distributed from
182
- `HuggingFaceTB/SmolLM2-1.7B`, which is released under the Apache-2.0
183
- license. No SmolLM2 model weights are included.
184
-
185
- ## Citation
186
-
187
- ```bibtex
188
- @misc{attnext2026,
189
- title = {AB-EXT 1.7B Binary16},
190
- author = {Anonymous Author(s)},
191
- year = {2026},
192
- url = {https://huggingface.co/E6E831728/AB-EXT-1.7B-Binary16}
193
- }
194
- ```