pszemraj SFconvertbot commited on
Commit
1722b64
·
verified ·
0 Parent(s):

Super-squash branch 'main' using huggingface_hub

Browse files

Co-authored-by: SFconvertbot <SFconvertbot@users.noreply.huggingface.co>

.gitattributes ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tflite filter=lfs diff=lfs merge=lfs -text
29
+ *.tgz filter=lfs diff=lfs merge=lfs -text
30
+ *.wasm filter=lfs diff=lfs merge=lfs -text
31
+ *.xz filter=lfs diff=lfs merge=lfs -text
32
+ *.zip filter=lfs diff=lfs merge=lfs -text
33
+ *.zst filter=lfs diff=lfs merge=lfs -text
34
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
README.md ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ - stable diffusion
5
+ - beautiful
6
+ - masterpiece
7
+ datasets:
8
+ - Gustavosta/Stable-Diffusion-Prompts
9
+ model-index:
10
+ - name: tiny-gpt2-magicprompt
11
+ results: []
12
+ widget:
13
+ - text: "morning sun over Jakarta"
14
+ example_title: "morning sun"
15
+ - text: "WARNING: pip is"
16
+ example_title: "pip"
17
+ - text: "sentient cheese"
18
+ example_title: "sentient cheese"
19
+ - text: "cheeps are"
20
+ example_title: "cheeps"
21
+ parameters:
22
+ min_length: 32
23
+ max_length: 64
24
+ no_repeat_ngram_size: 1
25
+ do_sample: True
26
+ ---
27
+
28
+
29
+ # tiny-gpt2-magicprompt
30
+
31
+
32
+ ~~Generate/augment your prompt, stable diffusion style.~~ Enter a new dimension of creativity
33
+
34
+ This model is a fine-tuned version of [sshleifer/tiny-gpt2](https://huggingface.co/sshleifer/tiny-gpt2) on the Gustavosta/Stable-Diffusion-Prompts dataset.
35
+ It achieves the following results on the evaluation set:
36
+ - Loss: 10.7918
37
+ - perplexity: 48618.8756
38
+
39
+ ## Intended uses & limitations
40
+
41
+ ???
42
+
43
+ ## Training and evaluation data
44
+
45
+ refer to the `Gustavosta/Stable-Diffusion-Prompts` dataset.
46
+
47
+ ## Training procedure
48
+
49
+ ### Training hyperparameters
50
+
51
+ The following hyperparameters were used during training:
52
+ - learning_rate: 0.0001
53
+ - train_batch_size: 8
54
+ - eval_batch_size: 2
55
+ - seed: 42
56
+ - distributed_type: multi-GPU
57
+ - num_devices: 2
58
+ - gradient_accumulation_steps: 32
59
+ - total_train_batch_size: 512
60
+ - total_eval_batch_size: 4
61
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
+ - lr_scheduler_type: cosine
63
+ - lr_scheduler_warmup_ratio: 0.05
64
+ - num_epochs: 10.0
65
+
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss |
69
+ |:-------------:|:-----:|:----:|:---------------:|
70
+ | 10.8201 | 0.96 | 16 | 10.8191 |
71
+ | 10.8167 | 1.96 | 32 | 10.8145 |
72
+ | 10.8117 | 2.96 | 48 | 10.8095 |
73
+ | 10.8058 | 3.96 | 64 | 10.8025 |
74
+ | 10.7997 | 4.96 | 80 | 10.7989 |
75
+ | 10.7959 | 5.96 | 96 | 10.7947 |
76
+ | 10.7934 | 6.96 | 112 | 10.7925 |
77
+ | 10.7924 | 7.96 | 128 | 10.7919 |
78
+ | 10.7921 | 8.96 | 144 | 10.7918 |
79
+ | 10.792 | 9.96 | 160 | 10.7918 |
80
+
81
+
82
+ ### Framework versions
83
+
84
+ - Transformers 4.25.0.dev0
85
+ - Pytorch 1.13.0+cu117
86
+ - Datasets 2.6.1
87
+ - Tokenizers 0.13.1
all_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.96,
3
+ "eval_loss": 10.791767120361328,
4
+ "eval_runtime": 5.99,
5
+ "eval_samples": 444,
6
+ "eval_samples_per_second": 74.123,
7
+ "eval_steps_per_second": 18.531,
8
+ "perplexity": 48618.875553525686,
9
+ "train_loss": 11.18234748840332,
10
+ "train_runtime": 256.0921,
11
+ "train_samples": 8516,
12
+ "train_samples_per_second": 332.537,
13
+ "train_steps_per_second": 0.625
14
+ }
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "sshleifer/tiny-gpt2",
3
+ "activation_function": "gelu_new",
4
+ "architectures": [
5
+ "GPT2LMHeadModel"
6
+ ],
7
+ "attn_pdrop": 0.1,
8
+ "bos_token_id": 50256,
9
+ "embd_pdrop": 0.1,
10
+ "eos_token_id": 50256,
11
+ "initializer_range": 0.02,
12
+ "layer_norm_epsilon": 1e-05,
13
+ "model_type": "gpt2",
14
+ "n_ctx": 1024,
15
+ "n_embd": 2,
16
+ "n_head": 2,
17
+ "n_inner": null,
18
+ "n_layer": 2,
19
+ "n_positions": 1024,
20
+ "reorder_and_upcast_attn": false,
21
+ "resid_pdrop": 0.1,
22
+ "scale_attn_by_inverse_layer_idx": false,
23
+ "scale_attn_weights": true,
24
+ "summary_activation": null,
25
+ "summary_first_dropout": 0.1,
26
+ "summary_proj_to_labels": true,
27
+ "summary_type": "cls_index",
28
+ "summary_use_proj": true,
29
+ "task_specific_params": {
30
+ "text-generation": {
31
+ "do_sample": true,
32
+ "no_repeat_ngram_size": 2,
33
+ "repetition_penalty": 4.5,
34
+ "max_length": 48
35
+ }
36
+ },
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.25.0.dev0",
39
+ "use_cache": true,
40
+ "vocab_size": 50257
41
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.96,
3
+ "eval_loss": 10.791767120361328,
4
+ "eval_runtime": 5.99,
5
+ "eval_samples": 444,
6
+ "eval_samples_per_second": 74.123,
7
+ "eval_steps_per_second": 18.531,
8
+ "perplexity": 48618.875553525686
9
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3dab14e92b6aed2ecce2f51fe5451cfeac14420af24a5d0ee8820964617851f4
3
+ size 2510830
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f524ede18c754cab273e7799daa996271fe23cdfef14fe89a01e2dad8ad8c8e
3
+ size 2517777
special_tokens_map.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "unk_token": "<|endoftext|>"
5
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_prefix_space": false,
4
+ "bos_token": {
5
+ "__type": "AddedToken",
6
+ "content": "<|endoftext|>",
7
+ "lstrip": false,
8
+ "normalized": true,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "eos_token": {
13
+ "__type": "AddedToken",
14
+ "content": "<|endoftext|>",
15
+ "lstrip": false,
16
+ "normalized": true,
17
+ "rstrip": false,
18
+ "single_word": false
19
+ },
20
+ "errors": "replace",
21
+ "model_max_length": 1024,
22
+ "name_or_path": "sshleifer/tiny-gpt2",
23
+ "pad_token": null,
24
+ "special_tokens_map_file": "/root/.cache/huggingface/hub/models--sshleifer--tiny-gpt2/snapshots/5f91d94bd9cd7190a9f3216ff93cd1dd95f2c7be/special_tokens_map.json",
25
+ "tokenizer_class": "GPT2Tokenizer",
26
+ "unk_token": {
27
+ "__type": "AddedToken",
28
+ "content": "<|endoftext|>",
29
+ "lstrip": false,
30
+ "normalized": true,
31
+ "rstrip": false,
32
+ "single_word": false
33
+ }
34
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.96,
3
+ "train_loss": 11.18234748840332,
4
+ "train_runtime": 256.0921,
5
+ "train_samples": 8516,
6
+ "train_samples_per_second": 332.537,
7
+ "train_steps_per_second": 0.625
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,297 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 9.962406015037594,
5
+ "global_step": 160,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.3,
12
+ "learning_rate": 6.25e-05,
13
+ "loss": 10.8217,
14
+ "step": 5
15
+ },
16
+ {
17
+ "epoch": 0.6,
18
+ "learning_rate": 9.995728791936504e-05,
19
+ "loss": 10.8213,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.9,
24
+ "learning_rate": 9.947761466636014e-05,
25
+ "loss": 10.8201,
26
+ "step": 15
27
+ },
28
+ {
29
+ "epoch": 0.96,
30
+ "eval_loss": 10.819129943847656,
31
+ "eval_runtime": 6.4186,
32
+ "eval_samples_per_second": 69.174,
33
+ "eval_steps_per_second": 17.294,
34
+ "step": 16
35
+ },
36
+ {
37
+ "epoch": 1.24,
38
+ "learning_rate": 9.847001329696653e-05,
39
+ "loss": 12.1716,
40
+ "step": 20
41
+ },
42
+ {
43
+ "epoch": 1.54,
44
+ "learning_rate": 9.694523495787149e-05,
45
+ "loss": 10.8181,
46
+ "step": 25
47
+ },
48
+ {
49
+ "epoch": 1.84,
50
+ "learning_rate": 9.491954909459895e-05,
51
+ "loss": 10.8167,
52
+ "step": 30
53
+ },
54
+ {
55
+ "epoch": 1.96,
56
+ "eval_loss": 10.81454086303711,
57
+ "eval_runtime": 6.3633,
58
+ "eval_samples_per_second": 69.775,
59
+ "eval_steps_per_second": 17.444,
60
+ "step": 32
61
+ },
62
+ {
63
+ "epoch": 2.18,
64
+ "learning_rate": 9.241456985587868e-05,
65
+ "loss": 12.1668,
66
+ "step": 35
67
+ },
68
+ {
69
+ "epoch": 2.48,
70
+ "learning_rate": 8.945702546981969e-05,
71
+ "loss": 10.8133,
72
+ "step": 40
73
+ },
74
+ {
75
+ "epoch": 2.78,
76
+ "learning_rate": 8.60784730526531e-05,
77
+ "loss": 10.8117,
78
+ "step": 45
79
+ },
80
+ {
81
+ "epoch": 2.96,
82
+ "eval_loss": 10.809527397155762,
83
+ "eval_runtime": 6.247,
84
+ "eval_samples_per_second": 71.074,
85
+ "eval_steps_per_second": 17.769,
86
+ "step": 48
87
+ },
88
+ {
89
+ "epoch": 3.12,
90
+ "learning_rate": 8.231496189304704e-05,
91
+ "loss": 12.1615,
92
+ "step": 50
93
+ },
94
+ {
95
+ "epoch": 3.42,
96
+ "learning_rate": 7.820664880476256e-05,
97
+ "loss": 10.8084,
98
+ "step": 55
99
+ },
100
+ {
101
+ "epoch": 3.72,
102
+ "learning_rate": 7.379736965185368e-05,
103
+ "loss": 10.8058,
104
+ "step": 60
105
+ },
106
+ {
107
+ "epoch": 3.96,
108
+ "eval_loss": 10.802533149719238,
109
+ "eval_runtime": 5.8936,
110
+ "eval_samples_per_second": 75.335,
111
+ "eval_steps_per_second": 18.834,
112
+ "step": 64
113
+ },
114
+ {
115
+ "epoch": 4.06,
116
+ "learning_rate": 6.91341716182545e-05,
117
+ "loss": 12.1538,
118
+ "step": 65
119
+ },
120
+ {
121
+ "epoch": 4.36,
122
+ "learning_rate": 6.426681121245527e-05,
123
+ "loss": 10.802,
124
+ "step": 70
125
+ },
126
+ {
127
+ "epoch": 4.66,
128
+ "learning_rate": 5.924722336357793e-05,
129
+ "loss": 10.8007,
130
+ "step": 75
131
+ },
132
+ {
133
+ "epoch": 4.96,
134
+ "learning_rate": 5.4128967273616625e-05,
135
+ "loss": 10.7997,
136
+ "step": 80
137
+ },
138
+ {
139
+ "epoch": 4.96,
140
+ "eval_loss": 10.798870086669922,
141
+ "eval_runtime": 6.2383,
142
+ "eval_samples_per_second": 71.174,
143
+ "eval_steps_per_second": 17.793,
144
+ "step": 80
145
+ },
146
+ {
147
+ "epoch": 5.3,
148
+ "learning_rate": 4.8966654938622295e-05,
149
+ "loss": 12.148,
150
+ "step": 85
151
+ },
152
+ {
153
+ "epoch": 5.6,
154
+ "learning_rate": 4.381536843653262e-05,
155
+ "loss": 10.7968,
156
+ "step": 90
157
+ },
158
+ {
159
+ "epoch": 5.9,
160
+ "learning_rate": 3.87300721992097e-05,
161
+ "loss": 10.7959,
162
+ "step": 95
163
+ },
164
+ {
165
+ "epoch": 5.96,
166
+ "eval_loss": 10.794721603393555,
167
+ "eval_runtime": 6.3176,
168
+ "eval_samples_per_second": 70.28,
169
+ "eval_steps_per_second": 17.57,
170
+ "step": 96
171
+ },
172
+ {
173
+ "epoch": 6.24,
174
+ "learning_rate": 3.3765026539765834e-05,
175
+ "loss": 12.1442,
176
+ "step": 100
177
+ },
178
+ {
179
+ "epoch": 6.54,
180
+ "learning_rate": 2.8973208692864624e-05,
181
+ "loss": 10.7943,
182
+ "step": 105
183
+ },
184
+ {
185
+ "epoch": 6.84,
186
+ "learning_rate": 2.4405747545519963e-05,
187
+ "loss": 10.7934,
188
+ "step": 110
189
+ },
190
+ {
191
+ "epoch": 6.96,
192
+ "eval_loss": 10.792530059814453,
193
+ "eval_runtime": 6.3054,
194
+ "eval_samples_per_second": 70.416,
195
+ "eval_steps_per_second": 17.604,
196
+ "step": 112
197
+ },
198
+ {
199
+ "epoch": 7.18,
200
+ "learning_rate": 2.0111378089837956e-05,
201
+ "loss": 12.1419,
202
+ "step": 115
203
+ },
204
+ {
205
+ "epoch": 7.48,
206
+ "learning_rate": 1.6135921418712956e-05,
207
+ "loss": 10.7926,
208
+ "step": 120
209
+ },
210
+ {
211
+ "epoch": 7.78,
212
+ "learning_rate": 1.2521795812943704e-05,
213
+ "loss": 10.7924,
214
+ "step": 125
215
+ },
216
+ {
217
+ "epoch": 7.96,
218
+ "eval_loss": 10.791926383972168,
219
+ "eval_runtime": 5.785,
220
+ "eval_samples_per_second": 76.75,
221
+ "eval_steps_per_second": 19.187,
222
+ "step": 128
223
+ },
224
+ {
225
+ "epoch": 8.12,
226
+ "learning_rate": 9.307564136490254e-06,
227
+ "loss": 12.1411,
228
+ "step": 130
229
+ },
230
+ {
231
+ "epoch": 8.42,
232
+ "learning_rate": 6.527522369181655e-06,
233
+ "loss": 10.7921,
234
+ "step": 135
235
+ },
236
+ {
237
+ "epoch": 8.72,
238
+ "learning_rate": 4.2113336672471245e-06,
239
+ "loss": 10.7921,
240
+ "step": 140
241
+ },
242
+ {
243
+ "epoch": 8.96,
244
+ "eval_loss": 10.79179859161377,
245
+ "eval_runtime": 6.2936,
246
+ "eval_samples_per_second": 70.548,
247
+ "eval_steps_per_second": 17.637,
248
+ "step": 144
249
+ },
250
+ {
251
+ "epoch": 9.06,
252
+ "learning_rate": 2.3837118562592797e-06,
253
+ "loss": 12.1411,
254
+ "step": 145
255
+ },
256
+ {
257
+ "epoch": 9.36,
258
+ "learning_rate": 1.064157733632276e-06,
259
+ "loss": 10.792,
260
+ "step": 150
261
+ },
262
+ {
263
+ "epoch": 9.66,
264
+ "learning_rate": 2.667509943378721e-07,
265
+ "loss": 10.792,
266
+ "step": 155
267
+ },
268
+ {
269
+ "epoch": 9.96,
270
+ "learning_rate": 0.0,
271
+ "loss": 10.792,
272
+ "step": 160
273
+ },
274
+ {
275
+ "epoch": 9.96,
276
+ "eval_loss": 10.791767120361328,
277
+ "eval_runtime": 5.8131,
278
+ "eval_samples_per_second": 76.379,
279
+ "eval_steps_per_second": 19.095,
280
+ "step": 160
281
+ },
282
+ {
283
+ "epoch": 9.96,
284
+ "step": 160,
285
+ "total_flos": 39596851200.0,
286
+ "train_loss": 11.18234748840332,
287
+ "train_runtime": 256.0921,
288
+ "train_samples_per_second": 332.537,
289
+ "train_steps_per_second": 0.625
290
+ }
291
+ ],
292
+ "max_steps": 160,
293
+ "num_train_epochs": 10,
294
+ "total_flos": 39596851200.0,
295
+ "trial_name": null,
296
+ "trial_params": null
297
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df9898973cea8c762a9cf27d5d9600cb8f1d87e0fa027798e8e32396be472be2
3
+ size 3387
vocab.json ADDED
The diff for this file is too large to render. See raw diff