Upload Chimera 510M at step 229000
Browse files- README.md +12 -7
- config.json +1 -1
- model.safetensors +1 -1
- training.log +0 -0
README.md
CHANGED
|
@@ -8,8 +8,11 @@ tags:
|
|
| 8 |
- hybrid-architecture
|
| 9 |
language:
|
| 10 |
- en
|
|
|
|
| 11 |
---
|
| 12 |
|
|
|
|
|
|
|
| 13 |
# Auron-510M
|
| 14 |
|
| 15 |
**Auron** — Chimera hybrid GDN-Attention language models with Ouroboros weight sharing.
|
|
@@ -28,12 +31,8 @@ language:
|
|
| 28 |
- **GDN:Attn ratio:** 3:1 (every 4th layer is attention)
|
| 29 |
- **Virtual equivalent:** ~1,020,434,560 params
|
| 30 |
|
| 31 |
-
## Training Curves
|
| 32 |
-
|
| 33 |
-

|
| 34 |
-
|
| 35 |
## Training
|
| 36 |
-
- **Step:**
|
| 37 |
- **Data:** Mixed (75% FineWeb-Edu, 18% StarCoder, 5% FineMath, 2% UltraChat)
|
| 38 |
- **Optimizer:** Muon + AdamW (decoupled embedding LR)
|
| 39 |
- **Schedule:** WSD (Warmup-Stable-Decay)
|
|
@@ -41,7 +40,7 @@ language:
|
|
| 41 |
## Usage
|
| 42 |
|
| 43 |
```bash
|
| 44 |
-
|
| 45 |
```
|
| 46 |
|
| 47 |
```python
|
|
@@ -53,6 +52,12 @@ generate(model, tokenizer, device, "The history of")
|
|
| 53 |
|
| 54 |
## Sampling
|
| 55 |
|
| 56 |
-
Default: T=0.7, top_k=20, top_p=0.9, rep_pen=1.0, presence_pen=1.5 (Ouroboros requires presence penalty >= 1.5 to prevent attractor wells).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
|
| 58 |
Built by [Florian Gasquez](https://fyx.jp) ([@nyxia](https://huggingface.co/nyxia)). Part of the [Soulkyn](https://soulkyn.com) project.
|
|
|
|
| 8 |
- hybrid-architecture
|
| 9 |
language:
|
| 10 |
- en
|
| 11 |
+
thumbnail: auron_banner.png
|
| 12 |
---
|
| 13 |
|
| 14 |
+

|
| 15 |
+
|
| 16 |
# Auron-510M
|
| 17 |
|
| 18 |
**Auron** — Chimera hybrid GDN-Attention language models with Ouroboros weight sharing.
|
|
|
|
| 31 |
- **GDN:Attn ratio:** 3:1 (every 4th layer is attention)
|
| 32 |
- **Virtual equivalent:** ~1,020,434,560 params
|
| 33 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 34 |
## Training
|
| 35 |
+
- **Step:** 229,000
|
| 36 |
- **Data:** Mixed (75% FineWeb-Edu, 18% StarCoder, 5% FineMath, 2% UltraChat)
|
| 37 |
- **Optimizer:** Muon + AdamW (decoupled embedding LR)
|
| 38 |
- **Schedule:** WSD (Warmup-Stable-Decay)
|
|
|
|
| 40 |
## Usage
|
| 41 |
|
| 42 |
```bash
|
| 43 |
+
git clone https://github.com/Fy-/Auron && cd Auron && rye sync
|
| 44 |
```
|
| 45 |
|
| 46 |
```python
|
|
|
|
| 52 |
|
| 53 |
## Sampling
|
| 54 |
|
| 55 |
+
Default: T=0.7, top_k=20, top_p=0.9, rep_pen=1.0, presence_pen=1.5 (Ouroboros weight sharing requires presence penalty >= 1.5 to prevent attractor wells).
|
| 56 |
+
|
| 57 |
+
## Links
|
| 58 |
+
|
| 59 |
+
- **Paper:** [Auron: Depth-Efficient Language Models via Hybrid Recurrent-Attention Weight Sharing](https://github.com/Fy-/Auron/blob/master/Auron_chimera_topology_paper.pdf)
|
| 60 |
+
- **Code:** [github.com/Fy-/Auron](https://github.com/Fy-/Auron)
|
| 61 |
+
- **Models:** [huggingface.co/nyxia](https://huggingface.co/nyxia)
|
| 62 |
|
| 63 |
Built by [Florian Gasquez](https://fyx.jp) ([@nyxia](https://huggingface.co/nyxia)). Part of the [Soulkyn](https://soulkyn.com) project.
|
config.json
CHANGED
|
@@ -26,7 +26,7 @@
|
|
| 26 |
"architecture": "Chimera",
|
| 27 |
"config_class": "ChimeraConfig",
|
| 28 |
"topology": "4 bottom + 4x3 top = 16 virtual",
|
| 29 |
-
"step":
|
| 30 |
"total_params": 510217280,
|
| 31 |
"size_label": "510M",
|
| 32 |
"model_type": "zara-ml"
|
|
|
|
| 26 |
"architecture": "Chimera",
|
| 27 |
"config_class": "ChimeraConfig",
|
| 28 |
"topology": "4 bottom + 4x3 top = 16 virtual",
|
| 29 |
+
"step": 229000,
|
| 30 |
"total_params": 510217280,
|
| 31 |
"size_label": "510M",
|
| 32 |
"model_type": "zara-ml"
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1487197656
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ed1d709dc0fbbe9152d1eb828d8c6010e380d0fdfd62dbf28bbf1332e97348b0
|
| 3 |
size 1487197656
|
training.log
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|