Model Details
Table with columns: Property, Value| Property | Value |
|---|
| Parameters | 15,735,168 |
| Architecture | LlamaForCausalLM |
| Layers | 8 |
| Hidden size | 384 |
| Attention heads | 6 |
| KV heads | 2 |
| Head dim | 64 |
| Intermediate size | 1024 |
| Vocabulary size | 8192 |
| Context length used in training | 2048 |
| Activation | SwiGLU / SiLU |
| Normalization | RMSNorm |
| Attention | GQA |
| Positional encoding | RoPE |
| Weight tying | Tied input embeddings and LM head |
| Training tokens | Approximately 30B |
| Training precision | bfloat16 |
| Optimizer | AdamW |
Tokenizer
Special tokens:
<|endoftext|>: 0
<|im_start|>: 1
<|im_end|>: 2
<|pad|>: 3
Training Data
The model was trained on a 30B-token pretraining mixture.
Stage 1 18,000,000,000 tokens 180 shards
Mixture:
dclm_baseline: 50%
finephrase: 20%
cosmopedia_v2: 10%
finemath_4plus: 10%
ultrafineweb_multistyle: 5%
ultrafineweb_qa: 5%
Stage 1.5 4,000,000,000 tokens 40 shards
This stage linearly transitions from the Stage 1 mixture to the Stage 2 mixture.
Stage 2 8,000,000,000 tokens 80 shards
Mixture:
finephrase: 30%
dclm_baseline: 30%
cosmopedia_v2: 18%
finemath_4plus: 10%
ultrafineweb_multistyle: 5%
ultrafineweb_qa: 5%
ultrachat: 2%
Training Summary
- Final step: 14,306
- Tokens seen: 30,000,000,000
- Tokens per step: 2,097,022
- Sequence length: 2048
- Last train loss: 2.8336
Usage
Notes on Generation
Veyra2-Mango-15M-Base is a raw base model. It is not instruction tuned and should not be expected to behave like a chat assistant.
Open-ended generations can be unstable, repetitive, or factually unreliable. It's not a polished assistant.
Intended Use
This model is intended for:
- small language model research
- continued pretraining
- benchmarking
- experimentation with compact causal LMs
Limitations
- Not instruction tuned
- Not RLHF tuned
- Not safe for factual or high-stakes use without additional validation
- Can hallucinate names, citations, species, references, and technical claims
- Open-ended text may drift off-topic
- Context length during training was 2048 tokens
Citation
If you use this model, please cite the model repository:
veyra-ai/Veyra2-Mango-15M-Base