Pebble-50M-beta / README.md
Hoglet-33's picture
Create README.md
3a25f23 verified
|
Raw History Blame Contribute Delete
2.35 kB
---
license: apache-2.0
language:
- en
pipeline_tag: text-generation
library_name: transformers
tags:
- pebble
- base-model
- mamba
- mamba2
- hybrid
- pytorch
- custom-architecture
---
# Pebble-50M-beta
Pebble-50M-beta is an experimental 50M-parameter language model designed to test how a larger Pebble architecture performs with a 16,384-token vocabulary and 16,384-token context window.
Despite having roughly twice the parameters of Pebble-25M, Pebble-50M-beta underperformed Pebble-25M and, on some evaluations, Pebble-10M. This model is therefore primarily useful as an experimental result rather than as the strongest Pebble model.
## Model Details
* **Architecture:** Hybrid Mamba2 / Transformer
* **Block Pattern:** 3 Mamba2 blocks : 1 Attention block (repeating)
* **Parameters:** ~49,334,448 (50M)
* **Hidden Dimension:** 768
* **Layers:** 8 (6 Mamba2, 2 Attention)
* **Vocab Size:** 16,384 (Custom Byte-Level BPE)
* **Context Length:** 16,384
* **Training Tokens:** ~25,000,000,000 (~25 Billion)
* **Optimizer:** Muon (for 2D hidden weights) + AdamW (for embeddings, norms, and scalars)
* **Precision:** fp32 master weights with bf16 autocast
## Dataset Sources
The model was trained on a 25B-token subset of the following datasets:
| Dataset | Token Allocation | Share |
| ------------- | ----------------: | -------: |
| FineWeb-Edu | 7.50 billion | 30% |
| DCLM | 5.00 billion | 20% |
| Cosmopedia-v2 | 3.75 billion | 15% |
| FineMath-4+ | 3.75 billion | 15% |
| FinePhrase | 3.00 billion | 12% |
| NPset | 2.00 billion | 8% |
| **Total** | **25.00 billion** | **100%** |
## Benchmarks
The original benchmark logs for this model were lost, so exact evaluation results are unavailable.
Qualitatively, Pebble-50M-beta underperformed Pebble-25M and, on some evaluations, Pebble-10M.
## Usage
Pebble-50M-beta does not require the `mamba-ssm` library and is intended to be usable with standard PyTorch-based inference implementations.
It may run on CUDA GPUs, AMD GPUs, Intel GPUs, and CPUs depending on the inference framework and available hardware acceleration.
## Status
This is a **beta/experimental model**. It is primarily intended for research and experimentation with the Pebble architecture.
## License
Apache 2.0