sovitrath commited on
Commit
e88af2b
·
verified ·
1 Parent(s): b578499

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - autotrain
4
+ - text-generation
5
+ library_name: transformers
6
+ widget:
7
+ - messages:
8
+ - role: user
9
+ content: What is your favorite condiment?
10
+ license: other
11
+ ---
12
+
13
+ # Model Trained Using AutoTrain
14
+
15
+ This model was trained using AutoTrain. For more information, please visit [AutoTrain](https://hf.co/docs/autotrain).
16
+
17
+ # Usage
18
+
19
+ ```python
20
+
21
+ from transformers import AutoModelForCausalLM, AutoTokenizer
22
+
23
+ model_path = "PATH_TO_THIS_REPO"
24
+
25
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
26
+ model = AutoModelForCausalLM.from_pretrained(
27
+ model_path,
28
+ device_map="auto",
29
+ torch_dtype='auto'
30
+ ).eval()
31
+
32
+ # Prompt content: "hi"
33
+ messages = [
34
+ {"role": "user", "content": "hi"}
35
+ ]
36
+
37
+ input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
38
+ output_ids = model.generate(input_ids.to('cuda'))
39
+ response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
40
+
41
+ # Model response: "Hello! How can I assist you today?"
42
+ print(response)
43
+ ```
checkpoint-902/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "openai-community/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": 768,
16
+ "n_head": 12,
17
+ "n_inner": null,
18
+ "n_layer": 12,
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
+ "max_length": 50
33
+ }
34
+ },
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.39.0",
37
+ "use_cache": false,
38
+ "vocab_size": 50257
39
+ }
checkpoint-902/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 50256,
4
+ "eos_token_id": 50256,
5
+ "transformers_version": "4.39.0"
6
+ }
checkpoint-902/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-902/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b60345a8949f01fb2e3b8c41ac0d063a751e7504f30226aa6ad403a88ec7a03b
3
+ size 497774208
checkpoint-902/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edbe4609ee3f3f62cdd4fc5e4fb976f9f376573545e3e3980658275f66c37f72
3
+ size 995642298
checkpoint-902/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9aebcec986913d86388e18c8304b17aed63fbdfd944d2519fdc774d4834d37c7
3
+ size 14244
checkpoint-902/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0023eb693df4b4ca689ed7e3bcd96faf58eddae57518ea202b0ad3b7a2cd0384
3
+ size 1064
checkpoint-902/special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
checkpoint-902/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-902/tokenizer_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "50256": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ }
12
+ },
13
+ "bos_token": "<|endoftext|>",
14
+ "chat_template": "{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
15
+ "clean_up_tokenization_spaces": true,
16
+ "eos_token": "<|endoftext|>",
17
+ "model_max_length": 1024,
18
+ "pad_token": "<|endoftext|>",
19
+ "tokenizer_class": "GPT2Tokenizer",
20
+ "unk_token": "<|endoftext|>"
21
+ }
checkpoint-902/trainer_state.json ADDED
@@ -0,0 +1,273 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.0,
5
+ "eval_steps": 500,
6
+ "global_step": 902,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.06,
13
+ "grad_norm": 2.5633599758148193,
14
+ "learning_rate": 5.494505494505495e-05,
15
+ "loss": 3.1831,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.11,
20
+ "grad_norm": 2.523214340209961,
21
+ "learning_rate": 0.0001098901098901099,
22
+ "loss": 2.9698,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.17,
27
+ "grad_norm": 2.6384470462799072,
28
+ "learning_rate": 0.00016483516483516484,
29
+ "loss": 2.7894,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.22,
34
+ "grad_norm": 2.2789688110351562,
35
+ "learning_rate": 0.00019778051787916153,
36
+ "loss": 2.7214,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.28,
41
+ "grad_norm": 1.6615029573440552,
42
+ "learning_rate": 0.00019161528976572133,
43
+ "loss": 2.7224,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.33,
48
+ "grad_norm": 1.7445863485336304,
49
+ "learning_rate": 0.00018545006165228113,
50
+ "loss": 2.7081,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.39,
55
+ "grad_norm": 1.5059261322021484,
56
+ "learning_rate": 0.00017928483353884094,
57
+ "loss": 2.668,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.44,
62
+ "grad_norm": 1.2297817468643188,
63
+ "learning_rate": 0.00017311960542540076,
64
+ "loss": 2.6149,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.5,
69
+ "grad_norm": 1.3275938034057617,
70
+ "learning_rate": 0.00016695437731196054,
71
+ "loss": 2.591,
72
+ "step": 225
73
+ },
74
+ {
75
+ "epoch": 0.55,
76
+ "grad_norm": 1.3491621017456055,
77
+ "learning_rate": 0.00016078914919852034,
78
+ "loss": 2.583,
79
+ "step": 250
80
+ },
81
+ {
82
+ "epoch": 0.61,
83
+ "grad_norm": 0.9846614003181458,
84
+ "learning_rate": 0.00015462392108508014,
85
+ "loss": 2.5189,
86
+ "step": 275
87
+ },
88
+ {
89
+ "epoch": 0.67,
90
+ "grad_norm": 0.9622399806976318,
91
+ "learning_rate": 0.00014845869297163997,
92
+ "loss": 2.549,
93
+ "step": 300
94
+ },
95
+ {
96
+ "epoch": 0.72,
97
+ "grad_norm": 0.9879335761070251,
98
+ "learning_rate": 0.00014229346485819977,
99
+ "loss": 2.4825,
100
+ "step": 325
101
+ },
102
+ {
103
+ "epoch": 0.78,
104
+ "grad_norm": 0.9367666840553284,
105
+ "learning_rate": 0.00013612823674475957,
106
+ "loss": 2.4431,
107
+ "step": 350
108
+ },
109
+ {
110
+ "epoch": 0.83,
111
+ "grad_norm": 0.8422412276268005,
112
+ "learning_rate": 0.00012996300863131935,
113
+ "loss": 2.399,
114
+ "step": 375
115
+ },
116
+ {
117
+ "epoch": 0.89,
118
+ "grad_norm": 1.3478554487228394,
119
+ "learning_rate": 0.00012379778051787915,
120
+ "loss": 2.4595,
121
+ "step": 400
122
+ },
123
+ {
124
+ "epoch": 0.94,
125
+ "grad_norm": 0.7269034385681152,
126
+ "learning_rate": 0.00011763255240443898,
127
+ "loss": 2.4464,
128
+ "step": 425
129
+ },
130
+ {
131
+ "epoch": 1.0,
132
+ "grad_norm": 0.9614003896713257,
133
+ "learning_rate": 0.00011146732429099878,
134
+ "loss": 2.474,
135
+ "step": 450
136
+ },
137
+ {
138
+ "epoch": 1.05,
139
+ "grad_norm": 0.8424810767173767,
140
+ "learning_rate": 0.00010530209617755857,
141
+ "loss": 2.3071,
142
+ "step": 475
143
+ },
144
+ {
145
+ "epoch": 1.11,
146
+ "grad_norm": 0.7542237639427185,
147
+ "learning_rate": 9.913686806411838e-05,
148
+ "loss": 2.2797,
149
+ "step": 500
150
+ },
151
+ {
152
+ "epoch": 1.16,
153
+ "grad_norm": 1.0208224058151245,
154
+ "learning_rate": 9.297163995067819e-05,
155
+ "loss": 2.2255,
156
+ "step": 525
157
+ },
158
+ {
159
+ "epoch": 1.22,
160
+ "grad_norm": 0.8640455007553101,
161
+ "learning_rate": 8.680641183723797e-05,
162
+ "loss": 2.3139,
163
+ "step": 550
164
+ },
165
+ {
166
+ "epoch": 1.27,
167
+ "grad_norm": 0.8734548687934875,
168
+ "learning_rate": 8.064118372379779e-05,
169
+ "loss": 2.321,
170
+ "step": 575
171
+ },
172
+ {
173
+ "epoch": 1.33,
174
+ "grad_norm": 0.7769907116889954,
175
+ "learning_rate": 7.447595561035759e-05,
176
+ "loss": 2.2748,
177
+ "step": 600
178
+ },
179
+ {
180
+ "epoch": 1.39,
181
+ "grad_norm": 0.7717509865760803,
182
+ "learning_rate": 6.831072749691739e-05,
183
+ "loss": 2.2641,
184
+ "step": 625
185
+ },
186
+ {
187
+ "epoch": 1.44,
188
+ "grad_norm": 0.8536700010299683,
189
+ "learning_rate": 6.214549938347719e-05,
190
+ "loss": 2.2658,
191
+ "step": 650
192
+ },
193
+ {
194
+ "epoch": 1.5,
195
+ "grad_norm": 0.8191765546798706,
196
+ "learning_rate": 5.5980271270037e-05,
197
+ "loss": 2.2729,
198
+ "step": 675
199
+ },
200
+ {
201
+ "epoch": 1.55,
202
+ "grad_norm": 1.0077537298202515,
203
+ "learning_rate": 4.9815043156596796e-05,
204
+ "loss": 2.2726,
205
+ "step": 700
206
+ },
207
+ {
208
+ "epoch": 1.61,
209
+ "grad_norm": 0.8226682543754578,
210
+ "learning_rate": 4.36498150431566e-05,
211
+ "loss": 2.2624,
212
+ "step": 725
213
+ },
214
+ {
215
+ "epoch": 1.66,
216
+ "grad_norm": 0.8827477693557739,
217
+ "learning_rate": 3.7484586929716406e-05,
218
+ "loss": 2.1979,
219
+ "step": 750
220
+ },
221
+ {
222
+ "epoch": 1.72,
223
+ "grad_norm": 0.8012568354606628,
224
+ "learning_rate": 3.131935881627621e-05,
225
+ "loss": 2.2563,
226
+ "step": 775
227
+ },
228
+ {
229
+ "epoch": 1.77,
230
+ "grad_norm": 0.8210929036140442,
231
+ "learning_rate": 2.5154130702836005e-05,
232
+ "loss": 2.2923,
233
+ "step": 800
234
+ },
235
+ {
236
+ "epoch": 1.83,
237
+ "grad_norm": 0.8134270906448364,
238
+ "learning_rate": 1.8988902589395807e-05,
239
+ "loss": 2.2719,
240
+ "step": 825
241
+ },
242
+ {
243
+ "epoch": 1.88,
244
+ "grad_norm": 0.8563103079795837,
245
+ "learning_rate": 1.282367447595561e-05,
246
+ "loss": 2.2821,
247
+ "step": 850
248
+ },
249
+ {
250
+ "epoch": 1.94,
251
+ "grad_norm": 0.8670592308044434,
252
+ "learning_rate": 6.6584463625154135e-06,
253
+ "loss": 2.2923,
254
+ "step": 875
255
+ },
256
+ {
257
+ "epoch": 2.0,
258
+ "grad_norm": 0.7817544341087341,
259
+ "learning_rate": 4.932182490752158e-07,
260
+ "loss": 2.2142,
261
+ "step": 900
262
+ }
263
+ ],
264
+ "logging_steps": 25,
265
+ "max_steps": 902,
266
+ "num_input_tokens_seen": 0,
267
+ "num_train_epochs": 2,
268
+ "save_steps": 500,
269
+ "total_flos": 5650178899968000.0,
270
+ "train_batch_size": 12,
271
+ "trial_name": null,
272
+ "trial_params": null
273
+ }
checkpoint-902/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b81fca1d626ce4e7d0f63d7ad9555d124639d9efeb4035c4a62b5d8d779a789d
3
+ size 4984
checkpoint-902/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "openai-community/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": 768,
16
+ "n_head": 12,
17
+ "n_inner": null,
18
+ "n_layer": 12,
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
+ "max_length": 50
33
+ }
34
+ },
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.39.0",
37
+ "use_cache": false,
38
+ "vocab_size": 50257
39
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 50256,
4
+ "eos_token_id": 50256,
5
+ "transformers_version": "4.39.0"
6
+ }
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:b60345a8949f01fb2e3b8c41ac0d063a751e7504f30226aa6ad403a88ec7a03b
3
+ size 497774208
special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "50256": {
5
+ "content": "<|endoftext|>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ }
12
+ },
13
+ "bos_token": "<|endoftext|>",
14
+ "chat_template": "{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
15
+ "clean_up_tokenization_spaces": true,
16
+ "eos_token": "<|endoftext|>",
17
+ "model_max_length": 1024,
18
+ "pad_token": "<|endoftext|>",
19
+ "tokenizer_class": "GPT2Tokenizer",
20
+ "unk_token": "<|endoftext|>"
21
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b81fca1d626ce4e7d0f63d7ad9555d124639d9efeb4035c4a62b5d8d779a789d
3
+ size 4984
training_params.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "openai-community/gpt2",
3
+ "project_name": "gpt2-base-open-assistant-guanaco",
4
+ "data_path": "timdettmers/openassistant-guanaco",
5
+ "train_split": "train",
6
+ "valid_split": null,
7
+ "add_eos_token": false,
8
+ "block_size": -1,
9
+ "model_max_length": 1024,
10
+ "padding": null,
11
+ "trainer": "default",
12
+ "use_flash_attention_2": false,
13
+ "log": "none",
14
+ "disable_gradient_checkpointing": false,
15
+ "logging_steps": -1,
16
+ "evaluation_strategy": "epoch",
17
+ "save_total_limit": 1,
18
+ "save_strategy": "epoch",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": null,
21
+ "lr": 0.0002,
22
+ "epochs": 2,
23
+ "batch_size": 12,
24
+ "warmup_ratio": 0.1,
25
+ "gradient_accumulation": 1,
26
+ "optimizer": "adamw_torch",
27
+ "scheduler": "linear",
28
+ "weight_decay": 0.0,
29
+ "max_grad_norm": 1.0,
30
+ "seed": 42,
31
+ "chat_template": null,
32
+ "quantization": null,
33
+ "target_modules": null,
34
+ "merge_adapter": false,
35
+ "peft": false,
36
+ "lora_r": 16,
37
+ "lora_alpha": 32,
38
+ "lora_dropout": 0.05,
39
+ "model_ref": null,
40
+ "dpo_beta": 0.1,
41
+ "prompt_text_column": "prompt",
42
+ "text_column": "text",
43
+ "rejected_text_column": "rejected",
44
+ "push_to_hub": true,
45
+ "repo_id": "sovitrath/gpt2_base_open_assistant_guanaco",
46
+ "username": null
47
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff