GGUFGuy commited on
Commit
b99addb
Β·
verified Β·
1 Parent(s): 22eb156

Create readme.md

Browse files
Files changed (1) hide show
  1. readme.md +242 -0
readme.md ADDED
@@ -0,0 +1,242 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+
3
+ language:
4
+ - en
5
+ library_name: transformers
6
+ pipeline_tag: text-generation
7
+ tags:
8
+ - novi
9
+ - novi-micro
10
+ - causal-lm
11
+ - from-scratch
12
+ - bananaall
13
+
14
+ ---
15
+
16
+ # Novi-Micro-Base
17
+
18
+ ![Novi-Micro Banner](banner.jpg)
19
+
20
+ **Novi-Micro-Base** is a tiny causal language model trained from scratch by **Novi-AI**.
21
+
22
+ With approximately **5.04 million parameters**, Novi-Micro explores language modeling at a small scale while using a substantially larger context window and training corpus than earlier Novi models.
23
+
24
+ ⚑ **5.04M parameters · 1B training tokens · 2,048-token context**
25
+
26
+ ## Model Details
27
+
28
+ ### Architecture
29
+
30
+ Novi-Micro-Base uses a custom **BananaMind 2-style decoder architecture** with RMSNorm, Rotary Position Embeddings (RoPE), grouped-query attention, QK normalization, SwiGLU feed-forward layers, and tied input/output embeddings.
31
+
32
+ | Property | Value |
33
+ | ------------------ | --------------------------------: |
34
+ | Model type | Causal Language Model |
35
+ | Architecture | **BananaMind 2-style** |
36
+ | Parameters | **5.04M** |
37
+ | Vocabulary size | **16,384** |
38
+ | Context length | **2,048 tokens** |
39
+ | Attention | **Grouped-Query Attention (GQA)** |
40
+ | Position encoding | **RoPE** |
41
+ | Normalization | **RMSNorm** |
42
+ | Feed-forward | **SwiGLU** |
43
+ | QK normalization | **Enabled** |
44
+ | Embeddings | **Tied** |
45
+ | Training precision | **FP32** |
46
+
47
+ ## Training
48
+
49
+ Novi-Micro-Base was trained from scratch for approximately **1 billion tokens**.
50
+
51
+ The training configuration used a **5M parameter target**, a sequence length of **2,048 tokens**, and a single dataset source.
52
+
53
+ ### Training Configuration
54
+
55
+ | Setting | Value |
56
+ | --------------------- | ----------------: |
57
+ | Training mode | **Pretraining** |
58
+ | Target size | **5M parameters** |
59
+ | Actual parameters | **5.04M** |
60
+ | Training tokens | **1,000,000,000** |
61
+ | Sequence length | **2,048** |
62
+ | Batch size | **2** |
63
+ | Gradient accumulation | **8** |
64
+ | Effective batch size | **16 sequences** |
65
+ | Learning rate | **2 Γ— 10⁻⁴** |
66
+ | Scheduler | **Cosine** |
67
+ | Warmup ratio | **3%** |
68
+ | Weight decay | **0.01** |
69
+ | Maximum gradient norm | **1.0** |
70
+ | Optimizer steps | **30,518** |
71
+ | Precision | **FP32** |
72
+ | Torch compile | **Enabled** |
73
+ | Random seed | **1337** |
74
+ | Checkpoint interval | **100 steps** |
75
+ | Logging interval | **10 steps** |
76
+
77
+ The model was trained until the selected **1,000,000,000-token target** was consumed.
78
+
79
+ ## Dataset
80
+
81
+ Novi-Micro-Base was trained using:
82
+
83
+ * **FineWeb-HQ**
84
+
85
+ The training configuration allocated a token target of exactly:
86
+
87
+ **1,000,000,000 tokens**
88
+
89
+ Documents were streamed from the dataset and packed into contiguous **2,048-token sequences** before being passed to the model.
90
+
91
+ ## Tokenizer
92
+
93
+ Novi-Micro uses a custom **byte-level BPE tokenizer** with a vocabulary size of **16,384 tokens**.
94
+
95
+ The tokenizer was trained from samples drawn from the training dataset before model pretraining.
96
+
97
+ Special tokens include:
98
+
99
+ * `[PAD]`
100
+ * `[BOS]`
101
+ * `[EOS]`
102
+ * `[UNK]`
103
+
104
+ The tokenizer uses ByteLevel pre-tokenization and decoding.
105
+
106
+ ## Architecture Details
107
+
108
+ Novi-Micro uses a decoder-only Transformer architecture.
109
+
110
+ ### Attention
111
+
112
+ The model uses **grouped-query attention (GQA)**, where multiple query heads share key and value heads.
113
+
114
+ Rotary Position Embeddings (**RoPE**) are applied to the query and key representations.
115
+
116
+ The BananaMind 2-style architecture also applies **RMSNorm to the query and key head dimensions**.
117
+
118
+ ### Feed-Forward Network
119
+
120
+ Each Transformer block uses a **SwiGLU** feed-forward network:
121
+
122
+ ```text
123
+ SwiGLU(x) = SiLU(gate(x)) Γ— up(x)
124
+ ```
125
+
126
+ The result is projected back to the model's hidden dimension.
127
+
128
+ ### Normalization
129
+
130
+ The model uses **RMSNorm** before both the attention and feed-forward sublayers.
131
+
132
+ ### Embeddings
133
+
134
+ The input token embeddings and language-model output embeddings are **tied**, reducing the number of independent parameters.
135
+
136
+ ## Intended Use
137
+
138
+ Novi-Micro-Base is primarily intended for:
139
+
140
+ * πŸ”¬ Research and experimentation
141
+ * πŸ§ͺ Small-model language-model experiments
142
+ * πŸŽ“ Educational purposes
143
+ * πŸ› οΈ Fine-tuning experiments
144
+ * πŸ’» Lightweight local inference
145
+ * πŸ€– Exploring language modeling at the million-parameter scale
146
+
147
+ As a **base model**, Novi-Micro-Base is not instruction-tuned and is not specifically trained to follow user commands or behave as a conversational assistant.
148
+
149
+ ## Limitations
150
+
151
+ Novi-Micro-Base is a very small experimental language model.
152
+
153
+ With approximately **5 million parameters**, it is dramatically smaller than modern general-purpose language models and should not be expected to match their capabilities.
154
+
155
+ It may:
156
+
157
+ * Generate incoherent text
158
+ * Repeat phrases
159
+ * Produce factual errors
160
+ * Struggle with complex instructions
161
+ * Have limited world knowledge
162
+ * Perform poorly on difficult reasoning tasks
163
+ * Produce grammatically unusual text
164
+ * Fail to maintain coherent long-form generations
165
+
166
+ A 2,048-token context window does not eliminate the limitations caused by the model's small parameter count.
167
+
168
+ This model should be considered a **research and experimentation model**, rather than a production-ready general-purpose LLM.
169
+
170
+ ## Usage
171
+
172
+ ```python
173
+ from transformers import AutoTokenizer, AutoModelForCausalLM
174
+
175
+ model_id = "Novi-AI/Novi-Micro-Base"
176
+
177
+ tokenizer = AutoTokenizer.from_pretrained(
178
+ model_id,
179
+ trust_remote_code=True,
180
+ )
181
+
182
+ model = AutoModelForCausalLM.from_pretrained(
183
+ model_id,
184
+ trust_remote_code=True,
185
+ )
186
+
187
+ prompt = "Hello, my name is"
188
+
189
+ inputs = tokenizer(prompt, return_tensors="pt")
190
+
191
+ outputs = model.generate(
192
+ **inputs,
193
+ max_new_tokens=50,
194
+ )
195
+
196
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
197
+ ```
198
+
199
+ Because Novi-Micro uses a custom architecture, `trust_remote_code=True` is required when loading the model through Transformers.
200
+
201
+ ## Project History
202
+
203
+ Novi-Micro-Base is part of **Project Kairo**, the development codename for the Novi model project.
204
+
205
+ The Novi series follows the earlier **AppleMind** experiments and represents the primary model-development line of Novi-AI.
206
+
207
+ **AppleMind β†’ Novi-Nano β†’ Novi-Micro β†’ future Novi models** πŸš€
208
+
209
+ Novi-Micro substantially expands upon Novi-Nano by increasing the model to approximately **5.04M parameters**, expanding the context window from 256 to **2,048 tokens**, and training on approximately **1 billion tokens**.
210
+
211
+ ## Training Infrastructure
212
+
213
+ The model was trained using GPU compute through the **BananaAll** training environment.
214
+
215
+ The training worker supports GPU acceleration through CUDA and Intel XPU, with Torch compilation enabled for supported environments.
216
+
217
+ This model was trained from scratch rather than fine-tuned from an existing language model checkpoint.
218
+
219
+ ## Acknowledgements
220
+
221
+ Novi-Micro was built using the open-source machine-learning ecosystem and datasets made available by the community.
222
+
223
+ Special thanks to:
224
+
225
+ * Hugging Face πŸ€—
226
+ * FineWeb
227
+ * FineWeb-HQ
228
+ * The open-source Transformers ecosystem
229
+
230
+ ## License
231
+
232
+ This model is released under the **Apache 2.0** license.
233
+
234
+ ---
235
+
236
+ ## 🧠 Novi AI
237
+
238
+ **Small models. Big experiments.**
239
+
240
+ Novi-Micro is intentionally tiny β€” exploring how far a language model can go with only a few million parameters and approximately one billion training tokens.
241
+
242
+ *Β© Novi AI 2026 β€” Project Kairo*