gyung commited on
Commit
17b2a84
·
verified ·
1 Parent(s): f380dda

Update model card for checkpoints/checkpoint-02B

Browse files
Files changed (1) hide show
  1. README.md +6 -6
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: 1,000,000,000 tokens
60
  - Sequence length: 4096 tokens
61
- - Global batch tokens: 524,288
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`