Hugging Face's logo

PrunaAI
/
distributed-optimized-gpt2-1b-bnb-8bit-smashed

Safetensors
gpt_optimized
pruna-ai
custom_code
8-bit precision
bitsandbytes
Model card Files Community
213
sharpenb commited on
Commit
39a0183
•
1 Parent(s): 8b7c475

e15d9973cded09c439ea503e0a0acbaae789afc70cc5639fbb382da1ba2c64e6

Browse files
Files changed (4) hide show
  1. README.md +2 -2
  2. config.json +311 -310
  3. model.safetensors +1 -1
  4. smash_config.json +2 -5
README.md CHANGED
@@ -14,8 +14,8 @@ tags:
14
  <!-- header start -->
15
  <!-- 200823 -->
16
  <div style="width: auto; margin-left: auto; margin-right: auto">
17
- <a href="https://www.pruna.ai/" target="_blank" rel="noopener noreferrer">
18
- <img src="https://i.imgur.com/eDAlcgk.png" alt="PrunaAI" style="width: 100%; min-width: 400px; display: block; margin: auto;">
19
  </a>
20
  </div>
21
  <!-- header end -->
 
14
  <!-- header start -->
15
  <!-- 200823 -->
16
  <div style="width: auto; margin-left: auto; margin-right: auto">
17
+ <a href="https://docs.pruna.ai/en/latest/setup/pip.html" target="_blank" rel="noopener noreferrer">
18
+ <img src="https://imgur.com/rVAgqMY.png" alt="PrunaAI" style="width: 100%; min-width: 400px; display: block; margin: auto;">
19
  </a>
20
  </div>
21
  <!-- header end -->
config.json CHANGED
@@ -1,312 +1,313 @@
1
  {
2
- "_name_or_path": "/covalent/.cache/models/tmpbjle89b1cw5xh7z9",
3
- "activation_function": "gelu_new",
4
- "all_reduce_scores": {
5
- "0": "NON_PARTICIPATING",
6
- "1": "NON_PARTICIPATING",
7
- "10": "SUCCESS",
8
- "100": "NON_PARTICIPATING",
9
- "101": "NON_PARTICIPATING",
10
- "102": "NON_PARTICIPATING",
11
- "103": "NON_PARTICIPATING",
12
- "104": "NON_PARTICIPATING",
13
- "105": "NON_PARTICIPATING",
14
- "106": "NON_PARTICIPATING",
15
- "107": "NON_PARTICIPATING",
16
- "108": "NON_PARTICIPATING",
17
- "109": "NON_PARTICIPATING",
18
- "11": "NON_PARTICIPATING",
19
- "110": "NON_PARTICIPATING",
20
- "111": "NON_PARTICIPATING",
21
- "112": "NON_PARTICIPATING",
22
- "113": "NON_PARTICIPATING",
23
- "114": "NON_PARTICIPATING",
24
- "115": "NON_PARTICIPATING",
25
- "116": "NON_PARTICIPATING",
26
- "117": "NON_PARTICIPATING",
27
- "118": "NON_PARTICIPATING",
28
- "119": "NON_PARTICIPATING",
29
- "12": "NON_PARTICIPATING",
30
- "120": "NON_PARTICIPATING",
31
- "121": "NON_PARTICIPATING",
32
- "122": "NON_PARTICIPATING",
33
- "123": "NON_PARTICIPATING",
34
- "124": "NON_PARTICIPATING",
35
- "125": "NON_PARTICIPATING",
36
- "126": "NON_PARTICIPATING",
37
- "127": "NON_PARTICIPATING",
38
- "128": "NON_PARTICIPATING",
39
- "129": "NON_PARTICIPATING",
40
- "13": "NON_PARTICIPATING",
41
- "130": "NON_PARTICIPATING",
42
- "131": "NON_PARTICIPATING",
43
- "132": "NON_PARTICIPATING",
44
- "133": "NON_PARTICIPATING",
45
- "134": "NON_PARTICIPATING",
46
- "135": "NON_PARTICIPATING",
47
- "136": "NON_PARTICIPATING",
48
- "137": "NON_PARTICIPATING",
49
- "138": "NON_PARTICIPATING",
50
- "139": "NON_PARTICIPATING",
51
- "14": "NON_PARTICIPATING",
52
- "140": "NON_PARTICIPATING",
53
- "141": "NON_PARTICIPATING",
54
- "142": "NON_PARTICIPATING",
55
- "143": "NON_PARTICIPATING",
56
- "144": "NON_PARTICIPATING",
57
- "145": "NON_PARTICIPATING",
58
- "146": "NON_PARTICIPATING",
59
- "147": "NON_PARTICIPATING",
60
- "148": "NON_PARTICIPATING",
61
- "149": "NON_PARTICIPATING",
62
- "15": "NON_PARTICIPATING",
63
- "150": "NON_PARTICIPATING",
64
- "151": "SUCCESS",
65
- "152": "NON_PARTICIPATING",
66
- "153": "NON_PARTICIPATING",
67
- "154": "NON_PARTICIPATING",
68
- "155": "NON_PARTICIPATING",
69
- "156": "SUCCESS",
70
- "157": "NON_PARTICIPATING",
71
- "158": "NON_PARTICIPATING",
72
- "159": "NON_PARTICIPATING",
73
- "16": "SUCCESS",
74
- "160": "NON_PARTICIPATING",
75
- "161": "NON_PARTICIPATING",
76
- "162": "NON_PARTICIPATING",
77
- "163": "NON_PARTICIPATING",
78
- "164": "NON_PARTICIPATING",
79
- "165": "NON_PARTICIPATING",
80
- "166": "NON_PARTICIPATING",
81
- "167": "NON_PARTICIPATING",
82
- "168": "NON_PARTICIPATING",
83
- "169": "NON_PARTICIPATING",
84
- "17": "NON_PARTICIPATING",
85
- "170": "NON_PARTICIPATING",
86
- "171": "NON_PARTICIPATING",
87
- "172": "NON_PARTICIPATING",
88
- "173": "NON_PARTICIPATING",
89
- "174": "NON_PARTICIPATING",
90
- "175": "NON_PARTICIPATING",
91
- "176": "NON_PARTICIPATING",
92
- "177": "NON_PARTICIPATING",
93
- "178": "NON_PARTICIPATING",
94
- "179": "NON_PARTICIPATING",
95
- "18": "NON_PARTICIPATING",
96
- "180": "NON_PARTICIPATING",
97
- "181": "NON_PARTICIPATING",
98
- "182": "NON_PARTICIPATING",
99
- "183": "NON_PARTICIPATING",
100
- "184": "NON_PARTICIPATING",
101
- "185": "NON_PARTICIPATING",
102
- "186": "NON_PARTICIPATING",
103
- "187": "NON_PARTICIPATING",
104
- "188": "NON_PARTICIPATING",
105
- "189": "NON_PARTICIPATING",
106
- "19": "NON_PARTICIPATING",
107
- "190": "NON_PARTICIPATING",
108
- "191": "NON_PARTICIPATING",
109
- "192": "NON_PARTICIPATING",
110
- "193": "NON_PARTICIPATING",
111
- "194": "NON_PARTICIPATING",
112
- "195": "NON_PARTICIPATING",
113
- "196": "NON_PARTICIPATING",
114
- "197": "NON_PARTICIPATING",
115
- "198": "NON_PARTICIPATING",
116
- "199": "NON_PARTICIPATING",
117
- "2": "NON_PARTICIPATING",
118
- "20": "NON_PARTICIPATING",
119
- "200": "NON_PARTICIPATING",
120
- "201": "NON_PARTICIPATING",
121
- "202": "NON_PARTICIPATING",
122
- "203": "NON_PARTICIPATING",
123
- "204": "NON_PARTICIPATING",
124
- "205": "NON_PARTICIPATING",
125
- "206": "NON_PARTICIPATING",
126
- "207": "NON_PARTICIPATING",
127
- "208": "NON_PARTICIPATING",
128
- "209": "SUCCESS",
129
- "21": "NON_PARTICIPATING",
130
- "210": "NON_PARTICIPATING",
131
- "211": "NON_PARTICIPATING",
132
- "212": "NON_PARTICIPATING",
133
- "213": "NON_PARTICIPATING",
134
- "214": "NON_PARTICIPATING",
135
- "215": "NON_PARTICIPATING",
136
- "216": "NON_PARTICIPATING",
137
- "217": "NON_PARTICIPATING",
138
- "218": "NON_PARTICIPATING",
139
- "219": "NON_PARTICIPATING",
140
- "22": "NON_PARTICIPATING",
141
- "220": "NON_PARTICIPATING",
142
- "221": "NON_PARTICIPATING",
143
- "222": "NON_PARTICIPATING",
144
- "223": "NON_PARTICIPATING",
145
- "224": "NON_PARTICIPATING",
146
- "225": "NON_PARTICIPATING",
147
- "226": "NON_PARTICIPATING",
148
- "227": "NON_PARTICIPATING",
149
- "228": "NON_PARTICIPATING",
150
- "229": "NON_PARTICIPATING",
151
- "23": "NON_PARTICIPATING",
152
- "230": "NON_PARTICIPATING",
153
- "231": "NON_PARTICIPATING",
154
- "232": "NON_PARTICIPATING",
155
- "233": "NON_PARTICIPATING",
156
- "234": "NON_PARTICIPATING",
157
- "235": "NON_PARTICIPATING",
158
- "236": "NON_PARTICIPATING",
159
- "237": "NON_PARTICIPATING",
160
- "238": "NON_PARTICIPATING",
161
- "239": "NON_PARTICIPATING",
162
- "24": "NON_PARTICIPATING",
163
- "240": "NON_PARTICIPATING",
164
- "241": "NON_PARTICIPATING",
165
- "242": "NON_PARTICIPATING",
166
- "243": "NON_PARTICIPATING",
167
- "244": "NON_PARTICIPATING",
168
- "245": "NON_PARTICIPATING",
169
- "246": "NON_PARTICIPATING",
170
- "247": "NON_PARTICIPATING",
171
- "248": "NON_PARTICIPATING",
172
- "249": "NON_PARTICIPATING",
173
- "25": "SUCCESS",
174
- "250": "NON_PARTICIPATING",
175
- "251": "NON_PARTICIPATING",
176
- "252": "NON_PARTICIPATING",
177
- "253": "NON_PARTICIPATING",
178
- "254": "NON_PARTICIPATING",
179
- "255": "NON_PARTICIPATING",
180
- "26": "NON_PARTICIPATING",
181
- "27": "SUCCESS",
182
- "28": "NON_PARTICIPATING",
183
- "29": "NON_PARTICIPATING",
184
- "3": "NON_PARTICIPATING",
185
- "30": "NON_PARTICIPATING",
186
- "31": "NON_PARTICIPATING",
187
- "32": "NON_PARTICIPATING",
188
- "33": "NON_PARTICIPATING",
189
- "34": "NON_PARTICIPATING",
190
- "35": "NON_PARTICIPATING",
191
- "36": "NON_PARTICIPATING",
192
- "37": "NON_PARTICIPATING",
193
- "38": "NON_PARTICIPATING",
194
- "39": "NON_PARTICIPATING",
195
- "4": "NON_PARTICIPATING",
196
- "40": "NON_PARTICIPATING",
197
- "41": "NON_PARTICIPATING",
198
- "42": "NON_PARTICIPATING",
199
- "43": "NON_PARTICIPATING",
200
- "44": "NON_PARTICIPATING",
201
- "45": "NON_PARTICIPATING",
202
- "46": "NON_PARTICIPATING",
203
- "47": "NON_PARTICIPATING",
204
- "48": "SUCCESS",
205
- "49": "NON_PARTICIPATING",
206
- "5": "NON_PARTICIPATING",
207
- "50": "NON_PARTICIPATING",
208
- "51": "NON_PARTICIPATING",
209
- "52": "SUCCESS",
210
- "53": "NON_PARTICIPATING",
211
- "54": "NON_PARTICIPATING",
212
- "55": "NON_PARTICIPATING",
213
- "56": "NON_PARTICIPATING",
214
- "57": "NON_PARTICIPATING",
215
- "58": "SUCCESS",
216
- "59": "NON_PARTICIPATING",
217
- "6": "NON_PARTICIPATING",
218
- "60": "NON_PARTICIPATING",
219
- "61": "NON_PARTICIPATING",
220
- "62": "NON_PARTICIPATING",
221
- "63": "NON_PARTICIPATING",
222
- "64": "NON_PARTICIPATING",
223
- "65": "NON_PARTICIPATING",
224
- "66": "NON_PARTICIPATING",
225
- "67": "NON_PARTICIPATING",
226
- "68": "NON_PARTICIPATING",
227
- "69": "NON_PARTICIPATING",
228
- "7": "NON_PARTICIPATING",
229
- "70": "NON_PARTICIPATING",
230
- "71": "NON_PARTICIPATING",
231
- "72": "NON_PARTICIPATING",
232
- "73": "NON_PARTICIPATING",
233
- "74": "NON_PARTICIPATING",
234
- "75": "NON_PARTICIPATING",
235
- "76": "NON_PARTICIPATING",
236
- "77": "NON_PARTICIPATING",
237
- "78": "SUCCESS",
238
- "79": "NON_PARTICIPATING",
239
- "8": "NON_PARTICIPATING",
240
- "80": "NON_PARTICIPATING",
241
- "81": "NON_PARTICIPATING",
242
- "82": "NON_PARTICIPATING",
243
- "83": "NON_PARTICIPATING",
244
- "84": "SUCCESS",
245
- "85": "NON_PARTICIPATING",
246
- "86": "NON_PARTICIPATING",
247
- "87": "NON_PARTICIPATING",
248
- "88": "NON_PARTICIPATING",
249
- "89": "NON_PARTICIPATING",
250
- "9": "NON_PARTICIPATING",
251
- "90": "NON_PARTICIPATING",
252
- "91": "NON_PARTICIPATING",
253
- "92": "NON_PARTICIPATING",
254
- "93": "NON_PARTICIPATING",
255
- "94": "NON_PARTICIPATING",
256
- "95": "NON_PARTICIPATING",
257
- "96": "NON_PARTICIPATING",
258
- "97": "NON_PARTICIPATING",
259
- "98": "NON_PARTICIPATING",
260
- "99": "NON_PARTICIPATING"
261
- },
262
- "architectures": [
263
- "GPTOptim"
264
- ],
265
- "attn_pdrop": 0.1,
266
- "auto_map": {
267
- "AutoConfig": "configuration_gpt_optimized.GPTOptimConfig",
268
- "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
269
- },
270
- "block_size": 1024,
271
- "bos_token_id": 50256,
272
- "embd_pdrop": 0.1,
273
- "eos_token_id": 50256,
274
- "initializer_range": 0.02,
275
- "layer_norm_epsilon": 1e-05,
276
- "model_type": "gpt_optimized",
277
- "n_embd": 1280,
278
- "n_head": 32,
279
- "n_inner": null,
280
- "n_layer": 48,
281
- "n_positions": 1024,
282
- "quantization_config": {
283
- "_load_in_4bit": false,
284
- "_load_in_8bit": true,
285
- "bnb_4bit_compute_dtype": "bfloat16",
286
- "bnb_4bit_quant_storage": "uint8",
287
- "bnb_4bit_quant_type": "fp4",
288
- "bnb_4bit_use_double_quant": false,
289
- "llm_int8_enable_fp32_cpu_offload": false,
290
- "llm_int8_has_fp16_weight": false,
291
- "llm_int8_skip_modules": [
292
- "lm_head"
293
  ],
294
- "llm_int8_threshold": 6.0,
295
- "load_in_4bit": false,
296
- "load_in_8bit": true,
297
- "quant_method": "bitsandbytes"
298
- },
299
- "reorder_and_upcast_attn": false,
300
- "resid_pdrop": 0.1,
301
- "scale_attn_by_inverse_layer_idx": false,
302
- "scale_attn_weights": true,
303
- "summary_activation": null,
304
- "summary_first_dropout": 0.1,
305
- "summary_proj_to_labels": true,
306
- "summary_type": "cls_index",
307
- "summary_use_proj": true,
308
- "torch_dtype": "float16",
309
- "transformers_version": "4.46.2",
310
- "use_cache": true,
311
- "vocab_size": 50257
312
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "_name_or_path": "/covalent/.cache/models/tmpr00129t8sa3uf5g5",
3
+ "activation_function": "gelu_new",
4
+ "all_reduce_scores": {
5
+ "0": "NON_PARTICIPATING",
6
+ "1": "NON_PARTICIPATING",
7
+ "10": "NON_PARTICIPATING",
8
+ "100": "NON_PARTICIPATING",
9
+ "101": "NON_PARTICIPATING",
10
+ "102": "NON_PARTICIPATING",
11
+ "103": "NON_PARTICIPATING",
12
+ "104": "NON_PARTICIPATING",
13
+ "105": "NON_PARTICIPATING",
14
+ "106": "NON_PARTICIPATING",
15
+ "107": "NON_PARTICIPATING",
16
+ "108": "NON_PARTICIPATING",
17
+ "109": "NON_PARTICIPATING",
18
+ "11": "NON_PARTICIPATING",
19
+ "110": "NON_PARTICIPATING",
20
+ "111": "NON_PARTICIPATING",
21
+ "112": "NON_PARTICIPATING",
22
+ "113": "NON_PARTICIPATING",
23
+ "114": "NON_PARTICIPATING",
24
+ "115": "NON_PARTICIPATING",
25
+ "116": "NON_PARTICIPATING",
26
+ "117": "NON_PARTICIPATING",
27
+ "118": "NON_PARTICIPATING",
28
+ "119": "NON_PARTICIPATING",
29
+ "12": "NON_PARTICIPATING",
30
+ "120": "NON_PARTICIPATING",
31
+ "121": "NON_PARTICIPATING",
32
+ "122": "NON_PARTICIPATING",
33
+ "123": "NON_PARTICIPATING",
34
+ "124": "NON_PARTICIPATING",
35
+ "125": "NON_PARTICIPATING",
36
+ "126": "NON_PARTICIPATING",
37
+ "127": "NON_PARTICIPATING",
38
+ "128": "NON_PARTICIPATING",
39
+ "129": "NON_PARTICIPATING",
40
+ "13": "NON_PARTICIPATING",
41
+ "130": "NON_PARTICIPATING",
42
+ "131": "NON_PARTICIPATING",
43
+ "132": "NON_PARTICIPATING",
44
+ "133": "NON_PARTICIPATING",
45
+ "134": "NON_PARTICIPATING",
46
+ "135": "NON_PARTICIPATING",
47
+ "136": "NON_PARTICIPATING",
48
+ "137": "NON_PARTICIPATING",
49
+ "138": "NON_PARTICIPATING",
50
+ "139": "NON_PARTICIPATING",
51
+ "14": "NON_PARTICIPATING",
52
+ "140": "NON_PARTICIPATING",
53
+ "141": "NON_PARTICIPATING",
54
+ "142": "SUCCESS",
55
+ "143": "NON_PARTICIPATING",
56
+ "144": "NON_PARTICIPATING",
57
+ "145": "NON_PARTICIPATING",
58
+ "146": "NON_PARTICIPATING",
59
+ "147": "NON_PARTICIPATING",
60
+ "148": "NON_PARTICIPATING",
61
+ "149": "NON_PARTICIPATING",
62
+ "15": "NON_PARTICIPATING",
63
+ "150": "NON_PARTICIPATING",
64
+ "151": "SUCCESS",
65
+ "152": "NON_PARTICIPATING",
66
+ "153": "NON_PARTICIPATING",
67
+ "154": "NON_PARTICIPATING",
68
+ "155": "NON_PARTICIPATING",
69
+ "156": "SUCCESS",
70
+ "157": "NON_PARTICIPATING",
71
+ "158": "SUCCESS",
72
+ "159": "SUCCESS",
73
+ "16": "NON_PARTICIPATING",
74
+ "160": "NON_PARTICIPATING",
75
+ "161": "NON_PARTICIPATING",
76
+ "162": "NON_PARTICIPATING",
77
+ "163": "NON_PARTICIPATING",
78
+ "164": "NON_PARTICIPATING",
79
+ "165": "NON_PARTICIPATING",
80
+ "166": "NON_PARTICIPATING",
81
+ "167": "NON_PARTICIPATING",
82
+ "168": "NON_PARTICIPATING",
83
+ "169": "NON_PARTICIPATING",
84
+ "17": "NON_PARTICIPATING",
85
+ "170": "NON_PARTICIPATING",
86
+ "171": "NON_PARTICIPATING",
87
+ "172": "NON_PARTICIPATING",
88
+ "173": "NON_PARTICIPATING",
89
+ "174": "NON_PARTICIPATING",
90
+ "175": "NON_PARTICIPATING",
91
+ "176": "NON_PARTICIPATING",
92
+ "177": "NON_PARTICIPATING",
93
+ "178": "SUCCESS",
94
+ "179": "NON_PARTICIPATING",
95
+ "18": "NON_PARTICIPATING",
96
+ "180": "NON_PARTICIPATING",
97
+ "181": "NON_PARTICIPATING",
98
+ "182": "NON_PARTICIPATING",
99
+ "183": "NON_PARTICIPATING",
100
+ "184": "NON_PARTICIPATING",
101
+ "185": "NON_PARTICIPATING",
102
+ "186": "NON_PARTICIPATING",
103
+ "187": "NON_PARTICIPATING",
104
+ "188": "NON_PARTICIPATING",
105
+ "189": "NON_PARTICIPATING",
106
+ "19": "NON_PARTICIPATING",
107
+ "190": "NON_PARTICIPATING",
108
+ "191": "NON_PARTICIPATING",
109
+ "192": "SUCCESS",
110
+ "193": "NON_PARTICIPATING",
111
+ "194": "NON_PARTICIPATING",
112
+ "195": "SUCCESS",
113
+ "196": "NON_PARTICIPATING",
114
+ "197": "NON_PARTICIPATING",
115
+ "198": "NON_PARTICIPATING",
116
+ "199": "NON_PARTICIPATING",
117
+ "2": "NON_PARTICIPATING",
118
+ "20": "NON_PARTICIPATING",
119
+ "200": "NON_PARTICIPATING",
120
+ "201": "NON_PARTICIPATING",
121
+ "202": "NON_PARTICIPATING",
122
+ "203": "NON_PARTICIPATING",
123
+ "204": "NON_PARTICIPATING",
124
+ "205": "NON_PARTICIPATING",
125
+ "206": "NON_PARTICIPATING",
126
+ "207": "NON_PARTICIPATING",
127
+ "208": "NON_PARTICIPATING",
128
+ "209": "NON_PARTICIPATING",
129
+ "21": "NON_PARTICIPATING",
130
+ "210": "NON_PARTICIPATING",
131
+ "211": "NON_PARTICIPATING",
132
+ "212": "NON_PARTICIPATING",
133
+ "213": "SUCCESS",
134
+ "214": "NON_PARTICIPATING",
135
+ "215": "NON_PARTICIPATING",
136
+ "216": "NON_PARTICIPATING",
137
+ "217": "SUCCESS",
138
+ "218": "NON_PARTICIPATING",
139
+ "219": "NON_PARTICIPATING",
140
+ "22": "NON_PARTICIPATING",
141
+ "220": "NON_PARTICIPATING",
142
+ "221": "NON_PARTICIPATING",
143
+ "222": "NON_PARTICIPATING",
144
+ "223": "NON_PARTICIPATING",
145
+ "224": "NON_PARTICIPATING",
146
+ "225": "NON_PARTICIPATING",
147
+ "226": "NON_PARTICIPATING",
148
+ "227": "NON_PARTICIPATING",
149
+ "228": "NON_PARTICIPATING",
150
+ "229": "NON_PARTICIPATING",
151
+ "23": "NON_PARTICIPATING",
152
+ "230": "NON_PARTICIPATING",
153
+ "231": "NON_PARTICIPATING",
154
+ "232": "NON_PARTICIPATING",
155
+ "233": "NON_PARTICIPATING",
156
+ "234": "NON_PARTICIPATING",
157
+ "235": "NON_PARTICIPATING",
158
+ "236": "NON_PARTICIPATING",
159
+ "237": "NON_PARTICIPATING",
160
+ "238": "NON_PARTICIPATING",
161
+ "239": "NON_PARTICIPATING",
162
+ "24": "NON_PARTICIPATING",
163
+ "240": "NON_PARTICIPATING",
164
+ "241": "NON_PARTICIPATING",
165
+ "242": "NON_PARTICIPATING",
166
+ "243": "NON_PARTICIPATING",
167
+ "244": "NON_PARTICIPATING",
168
+ "245": "NON_PARTICIPATING",
169
+ "246": "NON_PARTICIPATING",
170
+ "247": "NON_PARTICIPATING",
171
+ "248": "NON_PARTICIPATING",
172
+ "249": "NON_PARTICIPATING",
173
+ "25": "SUCCESS",
174
+ "250": "NON_PARTICIPATING",
175
+ "251": "NON_PARTICIPATING",
176
+ "252": "NON_PARTICIPATING",
177
+ "253": "NON_PARTICIPATING",
178
+ "254": "NON_PARTICIPATING",
179
+ "255": "NON_PARTICIPATING",
180
+ "26": "NON_PARTICIPATING",
181
+ "27": "NON_PARTICIPATING",
182
+ "28": "SUCCESS",
183
+ "29": "NON_PARTICIPATING",
184
+ "3": "NON_PARTICIPATING",
185
+ "30": "NON_PARTICIPATING",
186
+ "31": "NON_PARTICIPATING",
187
+ "32": "NON_PARTICIPATING",
188
+ "33": "NON_PARTICIPATING",
189
+ "34": "NON_PARTICIPATING",
190
+ "35": "NON_PARTICIPATING",
191
+ "36": "NON_PARTICIPATING",
192
+ "37": "NON_PARTICIPATING",
193
+ "38": "NON_PARTICIPATING",
194
+ "39": "NON_PARTICIPATING",
195
+ "4": "NON_PARTICIPATING",
196
+ "40": "NON_PARTICIPATING",
197
+ "41": "NON_PARTICIPATING",
198
+ "42": "NON_PARTICIPATING",
199
+ "43": "NON_PARTICIPATING",
200
+ "44": "NON_PARTICIPATING",
201
+ "45": "NON_PARTICIPATING",
202
+ "46": "NON_PARTICIPATING",
203
+ "47": "NON_PARTICIPATING",
204
+ "48": "NON_PARTICIPATING",
205
+ "49": "NON_PARTICIPATING",
206
+ "5": "NON_PARTICIPATING",
207
+ "50": "NON_PARTICIPATING",
208
+ "51": "NON_PARTICIPATING",
209
+ "52": "NON_PARTICIPATING",
210
+ "53": "NON_PARTICIPATING",
211
+ "54": "NON_PARTICIPATING",
212
+ "55": "SUCCESS",
213
+ "56": "NON_PARTICIPATING",
214
+ "57": "NON_PARTICIPATING",
215
+ "58": "NON_PARTICIPATING",
216
+ "59": "NON_PARTICIPATING",
217
+ "6": "NON_PARTICIPATING",
218
+ "60": "NON_PARTICIPATING",
219
+ "61": "NON_PARTICIPATING",
220
+ "62": "NON_PARTICIPATING",
221
+ "63": "NON_PARTICIPATING",
222
+ "64": "NON_PARTICIPATING",
223
+ "65": "NON_PARTICIPATING",
224
+ "66": "NON_PARTICIPATING",
225
+ "67": "NON_PARTICIPATING",
226
+ "68": "NON_PARTICIPATING",
227
+ "69": "NON_PARTICIPATING",
228
+ "7": "NON_PARTICIPATING",
229
+ "70": "NON_PARTICIPATING",
230
+ "71": "NON_PARTICIPATING",
231
+ "72": "NON_PARTICIPATING",
232
+ "73": "NON_PARTICIPATING",
233
+ "74": "NON_PARTICIPATING",
234
+ "75": "NON_PARTICIPATING",
235
+ "76": "NON_PARTICIPATING",
236
+ "77": "NON_PARTICIPATING",
237
+ "78": "SUCCESS",
238
+ "79": "NON_PARTICIPATING",
239
+ "8": "NON_PARTICIPATING",
240
+ "80": "NON_PARTICIPATING",
241
+ "81": "NON_PARTICIPATING",
242
+ "82": "NON_PARTICIPATING",
243
+ "83": "NON_PARTICIPATING",
244
+ "84": "NON_PARTICIPATING",
245
+ "85": "NON_PARTICIPATING",
246
+ "86": "NON_PARTICIPATING",
247
+ "87": "NON_PARTICIPATING",
248
+ "88": "SUCCESS",
249
+ "89": "NON_PARTICIPATING",
250
+ "9": "NON_PARTICIPATING",
251
+ "90": "NON_PARTICIPATING",
252
+ "91": "NON_PARTICIPATING",
253
+ "92": "NON_PARTICIPATING",
254
+ "93": "NON_PARTICIPATING",
255
+ "94": "SUCCESS",
256
+ "95": "NON_PARTICIPATING",
257
+ "96": "NON_PARTICIPATING",
258
+ "97": "NON_PARTICIPATING",
259
+ "98": "SUCCESS",
260
+ "99": "NON_PARTICIPATING"
261
+ },
262
+ "architectures": [
263
+ "GPTOptim"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
264
  ],
265
+ "attn_pdrop": 0.1,
266
+ "auto_map": {
267
+ "AutoConfig": "configuration_gpt_optimized.GPTOptimConfig",
268
+ "AutoModelForCausalLM": "distributed/optimized-gpt2-500m--modeling_gpt_optimized.GPTOptim"
269
+ },
270
+ "block_size": 1024,
271
+ "bos_token_id": 50256,
272
+ "embd_pdrop": 0.1,
273
+ "eos_token_id": 50256,
274
+ "initializer_range": 0.02,
275
+ "layer_norm_epsilon": 1e-05,
276
+ "model_type": "gpt_optimized",
277
+ "n_embd": 1280,
278
+ "n_head": 32,
279
+ "n_inner": null,
280
+ "n_layer": 48,
281
+ "n_positions": 1024,
282
+ "quantization_config": {
283
+ "_load_in_4bit": false,
284
+ "_load_in_8bit": true,
285
+ "bnb_4bit_compute_dtype": "bfloat16",
286
+ "bnb_4bit_quant_storage": "uint8",
287
+ "bnb_4bit_quant_type": "fp4",
288
+ "bnb_4bit_use_double_quant": false,
289
+ "llm_int8_enable_fp32_cpu_offload": false,
290
+ "llm_int8_has_fp16_weight": false,
291
+ "llm_int8_skip_modules": [
292
+ "lm_head"
293
+ ],
294
+ "llm_int8_threshold": 6.0,
295
+ "load_in_4bit": false,
296
+ "load_in_8bit": true,
297
+ "quant_method": "bitsandbytes"
298
+ },
299
+ "reorder_and_upcast_attn": false,
300
+ "resid_pdrop": 0.1,
301
+ "scale_attn_by_inverse_layer_idx": false,
302
+ "scale_attn_weights": true,
303
+ "summary_activation": null,
304
+ "summary_first_dropout": 0.1,
305
+ "summary_proj_to_labels": true,
306
+ "summary_type": "cls_index",
307
+ "summary_use_proj": true,
308
+ "torch_dtype": "float16",
309
+ "transformers_version": "4.46.2",
310
+ "use_cache": true,
311
+ "vocab_size": 50257,
312
+ "api_key": null
313
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ec07e7a8dfe90986cb227f5f8fb52b4ddbfac63dc029538d48d8e1c137be744a
3
  size 1207575528
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccccc2a01b558e8494a1433aa02aa5f7a3b6ac99923477f38e3fc8e74bee5fb3
3
  size 1207575528
smash_config.json CHANGED
@@ -28,11 +28,8 @@
28
  "quant_llm-int8_weight_bits": 8,
29
  "max_batch_size": 1,
30
  "device": "cuda",
31
- "cache_dir": "/covalent/.cache/models/tmpbjle89b1",
32
  "task": "",
33
  "save_load_fn": "bitsandbytes",
34
- "save_load_fn_args": {
35
- "weight_quantization_bits": "param.dtype"
36
- },
37
- "api_key": null
38
  }
 
28
  "quant_llm-int8_weight_bits": 8,
29
  "max_batch_size": 1,
30
  "device": "cuda",
31
+ "cache_dir": "/covalent/.cache/models/tmpr00129t8",
32
  "task": "",
33
  "save_load_fn": "bitsandbytes",
34
+ "save_load_fn_args": {}
 
 
 
35
  }