Update model card for checkpoints/checkpoint-02B
Browse files
README.md
CHANGED
|
@@ -16,10 +16,6 @@ datasets:
|
|
| 16 |
This repository stores milestone checkpoints for `gdn2_kla_1.3B` from the
|
| 17 |
`gdn2_kla_1.3B_fineweb_edu_10bt` run.
|
| 18 |
|
| 19 |
-
## Code
|
| 20 |
-
|
| 21 |
-
- GitHub: https://github.com/gyunggyung/Gated_Linear_Attention2
|
| 22 |
-
|
| 23 |
## What This Model Is
|
| 24 |
|
| 25 |
`gdn2_kla_1.3B` is a recurrent-only linear attention experiment. It starts from
|
|
@@ -40,6 +36,10 @@ k_t(\lambda_t w_t \odot v_t)^\top
|
|
| 40 |
It is not a standard Transformers checkpoint and does not use softmax attention
|
| 41 |
or SWA layers.
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
## License
|
| 44 |
|
| 45 |
The model weights in this Hugging Face repository are released under Apache-2.0.
|
|
@@ -56,9 +56,9 @@ model repository is intended to be used with the standalone Apache-2.0 runtime.
|
|
| 56 |
- Candidate: Kaczmarz-normalized GDN-2 gates
|
| 57 |
- Training data source: FineWeb-Edu `sample/100BT` local parquet
|
| 58 |
- Token budget for this run: 10B
|
| 59 |
-
- Current milestone:
|
| 60 |
- Sequence length: 4096 tokens
|
| 61 |
-
- Global batch tokens:
|
| 62 |
- Tokenizer: `TinyLlama/TinyLlama_v1.1`
|
| 63 |
- Data shuffle seed: `3407`
|
| 64 |
- Data shuffle buffer: `100000`
|
|
|
|
| 16 |
This repository stores milestone checkpoints for `gdn2_kla_1.3B` from the
|
| 17 |
`gdn2_kla_1.3B_fineweb_edu_10bt` run.
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
## What This Model Is
|
| 20 |
|
| 21 |
`gdn2_kla_1.3B` is a recurrent-only linear attention experiment. It starts from
|
|
|
|
| 36 |
It is not a standard Transformers checkpoint and does not use softmax attention
|
| 37 |
or SWA layers.
|
| 38 |
|
| 39 |
+
## Code
|
| 40 |
+
|
| 41 |
+
- GitHub: https://github.com/gyunggyung/Gated_Linear_Attention2
|
| 42 |
+
|
| 43 |
## License
|
| 44 |
|
| 45 |
The model weights in this Hugging Face repository are released under Apache-2.0.
|
|
|
|
| 56 |
- Candidate: Kaczmarz-normalized GDN-2 gates
|
| 57 |
- Training data source: FineWeb-Edu `sample/100BT` local parquet
|
| 58 |
- Token budget for this run: 10B
|
| 59 |
+
- Current milestone: 2,000,000,000 tokens
|
| 60 |
- Sequence length: 4096 tokens
|
| 61 |
+
- Global batch tokens: 1,048,576
|
| 62 |
- Tokenizer: `TinyLlama/TinyLlama_v1.1`
|
| 63 |
- Data shuffle seed: `3407`
|
| 64 |
- Data shuffle buffer: `100000`
|