nyxia commited on
Commit
65227d7
·
verified ·
1 Parent(s): 23ce10a

Upload Chimera 510M at step 229000

Browse files
Files changed (4) hide show
  1. README.md +12 -7
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. 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
- ![Training Curves](training_curves.png)
34
-
35
  ## Training
36
- - **Step:** 225,500
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
- pip install ouro
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
+ ![Auron](auron_banner.png)
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": 225500,
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:9bac19b2677ba1d1d5cf9f44b1f711f15ba0110a25227419d61ac213ffbda735
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