Muennighoff commited on
Commit
7540e91
·
verified ·
1 Parent(s): d6a9600

Training in progress, step 200

Browse files
Files changed (4) hide show
  1. README.md +7 -7
  2. config.json +3 -3
  3. model.safetensors +1 -1
  4. training_args.bin +2 -2
README.md CHANGED
@@ -3,8 +3,8 @@ library_name: transformers
3
  model_name: 1.5B-v70
4
  tags:
5
  - generated_from_trainer
6
- - grpo
7
  - trl
 
8
  licence: license
9
  ---
10
 
@@ -26,25 +26,25 @@ print(output["generated_text"])
26
 
27
  ## Training procedure
28
 
29
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/muennighoff/s2/runs/hsee6t4x)
30
 
31
 
32
  This model was trained with GRPO, a method introduced in [DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models](https://huggingface.co/papers/2402.03300).
33
 
34
  ### Framework versions
35
 
36
- - TRL: 0.22.0.dev0
37
- - Transformers: 4.55.4
38
- - Pytorch: 2.7.1
39
  - Datasets: 4.0.0
40
- - Tokenizers: 0.21.4
41
 
42
  ## Citations
43
 
44
  Cite GRPO as:
45
 
46
  ```bibtex
47
- @article{zhihong2024deepseekmath,
48
  title = {{DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models}},
49
  author = {Zhihong Shao and Peiyi Wang and Qihao Zhu and Runxin Xu and Junxiao Song and Mingchuan Zhang and Y. K. Li and Y. Wu and Daya Guo},
50
  year = 2024,
 
3
  model_name: 1.5B-v70
4
  tags:
5
  - generated_from_trainer
 
6
  - trl
7
+ - grpo
8
  licence: license
9
  ---
10
 
 
26
 
27
  ## Training procedure
28
 
29
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/muennighoff/s2/runs/w2qkgot7)
30
 
31
 
32
  This model was trained with GRPO, a method introduced in [DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models](https://huggingface.co/papers/2402.03300).
33
 
34
  ### Framework versions
35
 
36
+ - TRL: 0.23.0
37
+ - Transformers: 4.56.1
38
+ - Pytorch: 2.9.0.dev20250827+cu128
39
  - Datasets: 4.0.0
40
+ - Tokenizers: 0.22.0
41
 
42
  ## Citations
43
 
44
  Cite GRPO as:
45
 
46
  ```bibtex
47
+ @article{shao2024deepseekmath,
48
  title = {{DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models}},
49
  author = {Zhihong Shao and Peiyi Wang and Qihao Zhu and Runxin Xu and Junxiao Song and Mingchuan Zhang and Y. K. Li and Y. Wu and Daya Guo},
50
  year = 2024,
config.json CHANGED
@@ -4,7 +4,7 @@
4
  ],
5
  "attention_bias": false,
6
  "attention_dropout": 0.0,
7
- "bos_token_id": 151643,
8
  "eos_token_id": 151645,
9
  "head_dim": 128,
10
  "hidden_act": "silu",
@@ -47,13 +47,13 @@
47
  "num_attention_heads": 16,
48
  "num_hidden_layers": 28,
49
  "num_key_value_heads": 8,
 
50
  "rms_norm_eps": 1e-06,
51
  "rope_scaling": null,
52
  "rope_theta": 1000000,
53
  "sliding_window": 2048,
54
  "tie_word_embeddings": true,
55
- "torch_dtype": "bfloat16",
56
- "transformers_version": "4.55.4",
57
  "use_cache": false,
58
  "use_sliding_window": true,
59
  "vocab_size": 151936
 
4
  ],
5
  "attention_bias": false,
6
  "attention_dropout": 0.0,
7
+ "dtype": "bfloat16",
8
  "eos_token_id": 151645,
9
  "head_dim": 128,
10
  "hidden_act": "silu",
 
47
  "num_attention_heads": 16,
48
  "num_hidden_layers": 28,
49
  "num_key_value_heads": 8,
50
+ "pad_token_id": 151643,
51
  "rms_norm_eps": 1e-06,
52
  "rope_scaling": null,
53
  "rope_theta": 1000000,
54
  "sliding_window": 2048,
55
  "tie_word_embeddings": true,
56
+ "transformers_version": "4.56.1",
 
57
  "use_cache": false,
58
  "use_sliding_window": true,
59
  "vocab_size": 151936
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7b33577942909baf86426e784ef537737af0a75f2f89795c0f949ce41008e3e3
3
  size 4063515640
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d12a2b7919c43995d2ad6e1603ec1f1150f8a41a2d085f40c99f8acf5c4af8d2
3
  size 4063515640
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed933f9c05a8ada746dbb8c66c369b397f60efaae9c0c437ef3dacf8c75eb89a
3
- size 8785
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53ad4311ec5aa3db3dcbd4c3687f3b087eb8e2d880b7a90246033265773839b7
3
+ size 8977