Blancy commited on
Commit
5b6622e
·
verified ·
1 Parent(s): bbaaaf8

Model save

Browse files
README.md ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ model_name: Qwen3-0.6B-Open-R1-Distill
4
+ tags:
5
+ - generated_from_trainer
6
+ - trl
7
+ - sft
8
+ licence: license
9
+ ---
10
+
11
+ # Model Card for Qwen3-0.6B-Open-R1-Distill
12
+
13
+ This model is a fine-tuned version of [None](https://huggingface.co/None).
14
+ It has been trained using [TRL](https://github.com/huggingface/trl).
15
+
16
+ ## Quick start
17
+
18
+ ```python
19
+ from transformers import pipeline
20
+
21
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
22
+ generator = pipeline("text-generation", model="Blancy/Qwen3-0.6B-Open-R1-Distill", device="cuda")
23
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
24
+ print(output["generated_text"])
25
+ ```
26
+
27
+ ## Training procedure
28
+
29
+
30
+
31
+
32
+ This model was trained with SFT.
33
+
34
+ ### Framework versions
35
+
36
+ - TRL: 0.15.2
37
+ - Transformers: 4.52.3
38
+ - Pytorch: 2.5.1
39
+ - Datasets: 3.6.0
40
+ - Tokenizers: 0.21.1
41
+
42
+ ## Citations
43
+
44
+
45
+
46
+ Cite TRL as:
47
+
48
+ ```bibtex
49
+ @misc{vonwerra2022trl,
50
+ title = {{TRL: Transformer Reinforcement Learning}},
51
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallouédec},
52
+ year = 2020,
53
+ journal = {GitHub repository},
54
+ publisher = {GitHub},
55
+ howpublished = {\url{https://github.com/huggingface/trl}}
56
+ }
57
+ ```
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_flos": 5.211391058037965e+16,
3
+ "train_loss": 0.09458173938046872,
4
+ "train_runtime": 117.3732,
5
+ "train_samples": 1000,
6
+ "train_samples_per_second": 164.066,
7
+ "train_steps_per_second": 1.304
8
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 151645,
6
+ 151643
7
+ ],
8
+ "pad_token_id": 151643,
9
+ "temperature": 0.6,
10
+ "top_k": 20,
11
+ "top_p": 0.95,
12
+ "transformers_version": "4.52.3"
13
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a3dee87fff88c9f6799d6820e6ef32a6a7ff8a233b88314ff78247af8f2d3464
3
  size 1192135096
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eeb8dbfdbed5476a09608aaec303528b621ff55ca5cba3e6b83b39b35b189abd
3
  size 1192135096
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "total_flos": 5.211391058037965e+16,
3
+ "train_loss": 0.09458173938046872,
4
+ "train_runtime": 117.3732,
5
+ "train_samples": 1000,
6
+ "train_samples_per_second": 164.066,
7
+ "train_steps_per_second": 1.304
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,253 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 3.0,
6
+ "eval_steps": 500,
7
+ "global_step": 153,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.09803921568627451,
14
+ "grad_norm": 0.375,
15
+ "learning_rate": 2.5e-05,
16
+ "loss": 0.1074,
17
+ "step": 5
18
+ },
19
+ {
20
+ "epoch": 0.19607843137254902,
21
+ "grad_norm": 0.3359375,
22
+ "learning_rate": 4.9994719205415894e-05,
23
+ "loss": 0.099,
24
+ "step": 10
25
+ },
26
+ {
27
+ "epoch": 0.29411764705882354,
28
+ "grad_norm": 0.318359375,
29
+ "learning_rate": 4.981015154140181e-05,
30
+ "loss": 0.1039,
31
+ "step": 15
32
+ },
33
+ {
34
+ "epoch": 0.39215686274509803,
35
+ "grad_norm": 0.298828125,
36
+ "learning_rate": 4.936401774893088e-05,
37
+ "loss": 0.1032,
38
+ "step": 20
39
+ },
40
+ {
41
+ "epoch": 0.49019607843137253,
42
+ "grad_norm": 0.302734375,
43
+ "learning_rate": 4.8661548338815794e-05,
44
+ "loss": 0.1,
45
+ "step": 25
46
+ },
47
+ {
48
+ "epoch": 0.5882352941176471,
49
+ "grad_norm": 0.267578125,
50
+ "learning_rate": 4.771097912261625e-05,
51
+ "loss": 0.0995,
52
+ "step": 30
53
+ },
54
+ {
55
+ "epoch": 0.6862745098039216,
56
+ "grad_norm": 0.287109375,
57
+ "learning_rate": 4.652345465527847e-05,
58
+ "loss": 0.1007,
59
+ "step": 35
60
+ },
61
+ {
62
+ "epoch": 0.7843137254901961,
63
+ "grad_norm": 0.27734375,
64
+ "learning_rate": 4.511289757541801e-05,
65
+ "loss": 0.098,
66
+ "step": 40
67
+ },
68
+ {
69
+ "epoch": 0.8823529411764706,
70
+ "grad_norm": 0.271484375,
71
+ "learning_rate": 4.349584537511159e-05,
72
+ "loss": 0.0968,
73
+ "step": 45
74
+ },
75
+ {
76
+ "epoch": 0.9803921568627451,
77
+ "grad_norm": 0.2578125,
78
+ "learning_rate": 4.16912565129229e-05,
79
+ "loss": 0.0988,
80
+ "step": 50
81
+ },
82
+ {
83
+ "epoch": 1.0784313725490196,
84
+ "grad_norm": 0.279296875,
85
+ "learning_rate": 3.9720288143309735e-05,
86
+ "loss": 0.0961,
87
+ "step": 55
88
+ },
89
+ {
90
+ "epoch": 1.1764705882352942,
91
+ "grad_norm": 0.271484375,
92
+ "learning_rate": 3.7606048068332286e-05,
93
+ "loss": 0.0909,
94
+ "step": 60
95
+ },
96
+ {
97
+ "epoch": 1.2745098039215685,
98
+ "grad_norm": 0.30078125,
99
+ "learning_rate": 3.5373323819801494e-05,
100
+ "loss": 0.0942,
101
+ "step": 65
102
+ },
103
+ {
104
+ "epoch": 1.3725490196078431,
105
+ "grad_norm": 0.267578125,
106
+ "learning_rate": 3.304829204813215e-05,
107
+ "loss": 0.09,
108
+ "step": 70
109
+ },
110
+ {
111
+ "epoch": 1.4705882352941178,
112
+ "grad_norm": 0.271484375,
113
+ "learning_rate": 3.065821162505025e-05,
114
+ "loss": 0.0932,
115
+ "step": 75
116
+ },
117
+ {
118
+ "epoch": 1.5686274509803921,
119
+ "grad_norm": 0.255859375,
120
+ "learning_rate": 2.8231104058245068e-05,
121
+ "loss": 0.0894,
122
+ "step": 80
123
+ },
124
+ {
125
+ "epoch": 1.6666666666666665,
126
+ "grad_norm": 0.25390625,
127
+ "learning_rate": 2.579542496481177e-05,
128
+ "loss": 0.0964,
129
+ "step": 85
130
+ },
131
+ {
132
+ "epoch": 1.7647058823529411,
133
+ "grad_norm": 0.29296875,
134
+ "learning_rate": 2.3379730455158238e-05,
135
+ "loss": 0.0897,
136
+ "step": 90
137
+ },
138
+ {
139
+ "epoch": 1.8627450980392157,
140
+ "grad_norm": 0.26171875,
141
+ "learning_rate": 2.101234233871961e-05,
142
+ "loss": 0.0937,
143
+ "step": 95
144
+ },
145
+ {
146
+ "epoch": 1.9607843137254903,
147
+ "grad_norm": 0.279296875,
148
+ "learning_rate": 1.8721016076637528e-05,
149
+ "loss": 0.0909,
150
+ "step": 100
151
+ },
152
+ {
153
+ "epoch": 2.0588235294117645,
154
+ "grad_norm": 0.279296875,
155
+ "learning_rate": 1.6532615374355324e-05,
156
+ "loss": 0.0966,
157
+ "step": 105
158
+ },
159
+ {
160
+ "epoch": 2.156862745098039,
161
+ "grad_norm": 0.24609375,
162
+ "learning_rate": 1.4472797229233409e-05,
163
+ "loss": 0.0882,
164
+ "step": 110
165
+ },
166
+ {
167
+ "epoch": 2.2549019607843137,
168
+ "grad_norm": 0.251953125,
169
+ "learning_rate": 1.2565711125713656e-05,
170
+ "loss": 0.0869,
171
+ "step": 115
172
+ },
173
+ {
174
+ "epoch": 2.3529411764705883,
175
+ "grad_norm": 0.2578125,
176
+ "learning_rate": 1.0833715904694373e-05,
177
+ "loss": 0.0909,
178
+ "step": 120
179
+ },
180
+ {
181
+ "epoch": 2.450980392156863,
182
+ "grad_norm": 0.26171875,
183
+ "learning_rate": 9.297117626563687e-06,
184
+ "loss": 0.0902,
185
+ "step": 125
186
+ },
187
+ {
188
+ "epoch": 2.549019607843137,
189
+ "grad_norm": 0.23046875,
190
+ "learning_rate": 7.973931501207469e-06,
191
+ "loss": 0.091,
192
+ "step": 130
193
+ },
194
+ {
195
+ "epoch": 2.6470588235294117,
196
+ "grad_norm": 0.283203125,
197
+ "learning_rate": 6.879670676144916e-06,
198
+ "loss": 0.0914,
199
+ "step": 135
200
+ },
201
+ {
202
+ "epoch": 2.7450980392156863,
203
+ "grad_norm": 0.240234375,
204
+ "learning_rate": 6.027164359057668e-06,
205
+ "loss": 0.0909,
206
+ "step": 140
207
+ },
208
+ {
209
+ "epoch": 2.843137254901961,
210
+ "grad_norm": 0.2333984375,
211
+ "learning_rate": 5.426407407059619e-06,
212
+ "loss": 0.0888,
213
+ "step": 145
214
+ },
215
+ {
216
+ "epoch": 2.9411764705882355,
217
+ "grad_norm": 0.2431640625,
218
+ "learning_rate": 5.084443146135623e-06,
219
+ "loss": 0.0911,
220
+ "step": 150
221
+ },
222
+ {
223
+ "epoch": 3.0,
224
+ "step": 153,
225
+ "total_flos": 5.211391058037965e+16,
226
+ "train_loss": 0.09458173938046872,
227
+ "train_runtime": 117.3732,
228
+ "train_samples_per_second": 164.066,
229
+ "train_steps_per_second": 1.304
230
+ }
231
+ ],
232
+ "logging_steps": 5,
233
+ "max_steps": 153,
234
+ "num_input_tokens_seen": 0,
235
+ "num_train_epochs": 3,
236
+ "save_steps": 100,
237
+ "stateful_callbacks": {
238
+ "TrainerControl": {
239
+ "args": {
240
+ "should_epoch_stop": false,
241
+ "should_evaluate": false,
242
+ "should_log": false,
243
+ "should_save": true,
244
+ "should_training_stop": true
245
+ },
246
+ "attributes": {}
247
+ }
248
+ },
249
+ "total_flos": 5.211391058037965e+16,
250
+ "train_batch_size": 128,
251
+ "trial_name": null,
252
+ "trial_params": null
253
+ }