LH-Tech-AI commited on
Commit
fee2e4c
·
verified ·
1 Parent(s): 56849e3

Delete README_final.md

Browse files
Files changed (1) hide show
  1. README_final.md +0 -209
README_final.md DELETED
@@ -1,209 +0,0 @@
1
- ---
2
- license: apache-2.0
3
- viewer: false
4
- datasets:
5
- - HuggingFaceFW/fineweb-edu
6
- - HuggingFaceFW/dclm_100BT-shuffled
7
- language:
8
- - en
9
- pipeline_tag: text-generation
10
- library_name: transformers
11
- tags:
12
- - small
13
- - supra
14
- - supra2
15
- - qwen3
16
- - sota
17
- ---
18
-
19
- <h1 align="center">Supra2-100M Base</h1>
20
-
21
- <p align="center">
22
- Base model of new Supra2 family • 100M Parameters • 1K Context - extended to 2k
23
- </p>
24
-
25
- ![supra2-_pro_-100m](https://cdn-uploads.huggingface.co/production/uploads/697f2832c2c5e4daa93cece7/zMBPPr0y634MNUvx2Krqy.png)
26
-
27
- **Supra2-100M Base** is a 100M-parameter decoder-only language model pretrained from scratch by **SupraLabs** on 30B tokens of English web text. It uses the **Qwen3** architecture with a custom 32,768-token tokenizer.
28
-
29
- This is a **base model**. It has *not* been instruction-tuned, chat-tuned, or aligned in any way.
30
-
31
- ---
32
-
33
- ## **Let the model speak - a sample**
34
-
35
- Prompt: "Artificial intelligence is "
36
- Completion:
37
- ```plaintext
38
- 100% efficient, yet it still requires a lot of resources.
39
- We are in the midst of a major transition from the traditional computer and software to artificial intelligence.
40
- It is very important to have a strategy in mind. This is not only important for the companies that use it, but also for individuals and organizations that use it.
41
- There is a huge difference between the two approaches. The artificial intelligence approach is based on computers and the software.
42
- It has been proven that the artificial intelligence approach is more efficient and has a much better performance.
43
- The benefits of artificial intelligence are also known to be immense. It is a powerful tool that can improve productivity, increase efficiency, and reduce costs.
44
- It can be used for a wide range of tasks, from customer service to product development.
45
- It is also used for various other tasks, such as sales, marketing, and customer service.
46
- It is also used for various other tasks, such as customer service and product development.
47
- There are a number of different types of artificial intelligence. These include machine learning, natural language processing, and neural networks.
48
- The main advantage of using artificial intelligence is that it is more efficient and can be used for a wide range of tasks.
49
- It is also[...]
50
- ```
51
-
52
- ---
53
-
54
- ## **Evaluation & Benchmarks**
55
-
56
- All benchmarks were evaluated using the EleutherAI LM-Eval Harness.
57
-
58
- | Model | PIQA (acc_norm) | HellaSwag (acc_norm) | ARC-Easy (acc_norm) | ARC-Challenge (acc_norm) |
59
- | :--- | :---: | :---: | :---: | :---: |
60
- | Supra-50M-Base | 0.62 | 0.32 | 0.46 | 0.25 |
61
- | Supra2-100M Base | 0.65 | 0.36 | 0.48 | 0.25 |
62
- | BananaMind-2-Pro-Preview-EXP | 0.67 | 0.40 | 0.51 | 0.27 |
63
- | GPT-X-125M | 0.65 | 0.37 | 0.51 | 0.25 |
64
- | OpenAI GPT-2 Small | 0.62 | 0.31 | 0.39 | 0.22 |
65
-
66
- ![grafik](https://cdn-uploads.huggingface.co/production/uploads/697f2832c2c5e4daa93cece7/TfH9iBYB0nNvms2-M84lG.png)
67
-
68
- ---
69
-
70
- ## **Model Details**
71
-
72
- | | |
73
- | ----- | ----- |
74
- | **Developed by** | SupraLabs |
75
- | **Model type** | Causal decoder-only transformer (Qwen3) |
76
- | **Language** | English |
77
- | **Parameters** | 100.7M total / 75.5M non-embedding |
78
- | **Training tokens** | 30B (300 tokens per parameter) |
79
- | **Context length** | 1,024 (trained) / 2,048 (config maximum, untested) |
80
- | **Precision** | bfloat16 |
81
- | **License** | Apache 2.0 |
82
-
83
- ### **Architecture**
84
-
85
- | Hyperparameter | Value |
86
- | ----- | ----- |
87
- | Hidden size | 768 |
88
- | Layers | 12 |
89
- | Attention heads | 12 (GQA, 4 KV heads) |
90
- | Head dim | 64 |
91
- | Intermediate size (SwiGLU) | 2048 |
92
- | Vocab size | 32,768 |
93
- | Positional encoding | RoPE θ=10,000 |
94
- | Normalization | RMSNorm, ε=10\-6 (incl. QK-Norm) |
95
- | Tied embeddings | Yes |
96
- | Sliding window | Disabled |
97
- | Attention implementation | SDPA |
98
-
99
- ---
100
-
101
- ## **Training Data**
102
-
103
- | Source | Share | Approx. tokens |
104
- | ----- | ----- | ----- |
105
- | `HuggingFaceFW/fineweb-edu` (`sample-350BT`) | 70% | 21B |
106
- | `HuggingFaceFW/dclm_100BT-shuffled` | 30% | 9B |
107
-
108
- Documents were tokenized with the custom `supra2-tokenizer`, concatenated into a flat `uint16` token stream, and packed into contiguous 1,024-token chunks (no padding, no document masking — sequences may cross document boundaries).
109
-
110
- ---
111
-
112
- ## **Training Procedure**
113
-
114
- | Setting | Value |
115
- | ----- | ----- |
116
- | Optimizer | AdamW (fused), β1=0.9, β2=0.95, ε=10−8 |
117
- | Peak learning rate | 1×10-3 |
118
- | LR schedule | WSD (Warmup–Stable–Decay), `1-sqrt` decay to 0 |
119
- | Total steps | 114k |
120
- | Micro batch size | 16 |
121
- | Gradient accumulation | 16 |
122
- | Effective batch | 256 sequences \= **262,144 tokens/step** |
123
- | Weight decay | 0.1 |
124
- | Gradient clipping | 1.0 |
125
- | Auxiliary loss | Router-free **z-loss**, coefficient 10\-4 |
126
- | Compile | `torch.compile` enabled |
127
- | Hardware | 1× RTX 5090 (32GB) |
128
-
129
- ---
130
-
131
- ## **Usage**
132
-
133
- ```python
134
- import torch
135
- from transformers import AutoModelForCausalLM, AutoTokenizer
136
-
137
- model_id = "SupraLabs/Supra2-100M"
138
-
139
- tokenizer = AutoTokenizer.from_pretrained(model_id)
140
- model = AutoModelForCausalLM.from_pretrained(
141
- model_id,
142
- torch_dtype=torch.bfloat16,
143
- device_map="auto",
144
- )
145
- model.eval()
146
-
147
- prompt = "Artificial intelligence is "
148
- inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
149
-
150
- with torch.no_grad():
151
- out = model.generate(
152
- **inputs,
153
- max_new_tokens=256,
154
- do_sample=True,
155
- temperature=0.3,
156
- top_p=0.85,
157
- top_k=15,
158
- no_repeat_ngram_size=3,
159
- repetition_penalty=1.1,
160
- )
161
-
162
- print(tokenizer.decode(out[0], skip_special_tokens=True))
163
- ```
164
-
165
- ### **Tokenizer notes**
166
-
167
- The tokenizer ships with a **ChatML** template and reserved special tokens (`<|im_start|>`, `<|im_end|>`, `<think>`, `<tool_call>`, `<|reserved_0..15|>`, …). These exist so that downstream fine-tunes can adopt them without resizing embeddings — **the base model has never seen them in training** and will not behave sensibly if you apply the chat template.
168
-
169
- ---
170
-
171
- ## **Intended Use**
172
-
173
- **Intended:**
174
-
175
- * Research on small-scale pretraining, scaling behavior, and data mixtures
176
- * A cheap starting point for SFT / instruction tuning / domain adaptation
177
- * Educational use, ablations, and fast iteration on limited hardware
178
-
179
- **Not intended:**
180
-
181
- * Any production or user-facing deployment
182
- * Factual question answering, advice, or decision support
183
- * Non-English text (essentially untrained; only supports english\!)
184
-
185
- ---
186
-
187
- ## **Limitations and Bias**
188
-
189
- * **Small.** At 100M parameters, factual recall, reasoning, arithmetic, and long-range coherence are weak. Expect frequent hallucination and topic drift.
190
- * **Base model.** No RLHF, no safety tuning, no refusal behavior. It will continue whatever text you give it, including harmful or offensive prompts.
191
- * **Web-derived data.** FineWeb-Edu and DCLM are filtered CommonCrawl derivatives and carry the biases, stereotypes, and factual errors of the open web.
192
- * **Short context.** Trained exclusively at 1,024 tokens. The config allows 2,048, but extrapolation beyond 1,024 is untested and likely degraded.
193
- * **No document masking.** Attention could cross document boundaries within a packed chunk, which slightly blurs document independence.
194
- * **English only.**
195
-
196
- ---
197
-
198
- ## **What comes next?**
199
-
200
- We are already training the next models in the Supra2-family:
201
-
202
- - Supra2-Nano (800k params)
203
- - Supra2-Small (\~1.3M params)
204
- - Supra2-Medium (25M params)
205
- - Supra2-IMG: our SOTA small text-to-image model
206
-
207
- ---
208
-
209
- *© SupraLabs 2026*