File size: 6,861 Bytes
03ae413
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
{
  "architectures": [
    "LagunaForCausalLM"
  ],
  "auto_map": {
    "AutoConfig": "configuration_laguna.LagunaConfig",
    "AutoModelForCausalLM": "modeling_laguna.LagunaForCausalLM"
  },
  "model_type": "laguna",
  "vocab_size": 100352,
  "hidden_size": 2048,
  "intermediate_size": 8192,
  "num_hidden_layers": 40,
  "num_attention_heads": 48,
  "num_key_value_heads": 8,
  "head_dim": 128,
  "max_position_embeddings": 131072,
  "attention_bias": false,
  "attention_dropout": 0.0,
  "rms_norm_eps": 1e-06,
  "num_experts": 256,
  "num_experts_per_tok": 8,
  "moe_intermediate_size": 512,
  "shared_expert_intermediate_size": 512,
  "router_aux_loss_coef": 0.0,
  "bos_token_id": 2,
  "eos_token_id": [
    2,
    24
  ],
  "pad_token_id": 9,
  "tie_word_embeddings": false,
  "use_cache": true,
  "torch_dtype": "bfloat16",
  "gating": true,
  "sliding_window": 512,
  "rope_parameters": {
    "full_attention": {
      "rope_theta": 500000.0,
      "rope_type": "yarn",
      "factor": 32.0,
      "original_max_position_embeddings": 4096,
      "beta_slow": 1.0,
      "beta_fast": 64.0,
      "attention_factor": 1.0,
      "partial_rotary_factor": 0.5
    },
    "sliding_attention": {
      "rope_type": "default",
      "rope_theta": 10000.0,
      "partial_rotary_factor": 1.0
    },
    "original_max_position_embeddings": 4096
  },
  "layer_types": [
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention",
    "full_attention",
    "sliding_attention",
    "sliding_attention",
    "sliding_attention"
  ],
  "moe_apply_router_weight_on_input": false,
  "partial_rotary_factor": 0.5,
  "mlp_layer_types": [
    "dense",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse",
    "sparse"
  ],
  "use_bidirectional_attention": false,
  "moe_routed_scaling_factor": 2.5,
  "num_attention_heads_per_layer": [
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64,
    48,
    64,
    64,
    64
  ],
  "quantization_config": {
    "config_groups": {
      "group_0": {
        "format": "pack-quantized",
        "input_activations": null,
        "output_activations": null,
        "targets": [
          "re:.*layers\\.([1-9]|[12]\\d|30)\\..*(w[1-3]|gate_proj|up_proj|down_proj)$"
        ],
        "weights": {
          "actorder": null,
          "block_structure": null,
          "dynamic": false,
          "group_size": 128,
          "num_bits": 4,
          "observer": "memoryless_minmax",
          "observer_kwargs": {},
          "scale_dtype": null,
          "strategy": "group",
          "symmetric": true,
          "type": "int",
          "zp_dtype": null
        }
      },
      "group_1": {
        "format": "pack-quantized",
        "input_activations": null,
        "output_activations": null,
        "targets": [
          "re:.*layers\\.3[1-9]\\..*(w[1-3]|gate_proj|up_proj|down_proj)$"
        ],
        "weights": {
          "actorder": null,
          "block_structure": null,
          "dynamic": false,
          "group_size": 128,
          "num_bits": 8,
          "observer": "memoryless_minmax",
          "observer_kwargs": {},
          "scale_dtype": null,
          "strategy": "group",
          "symmetric": true,
          "type": "int",
          "zp_dtype": null
        }
      }
    },
    "format": "pack-quantized",
    "global_compression_ratio": null,
    "ignore": [
      "lm_head",
      "model.layers.0.mlp.gate_proj",
      "model.layers.0.mlp.up_proj",
      "model.layers.0.mlp.down_proj",
      "re:.*\\.self_attn\\.q_proj$",
      "re:.*\\.self_attn\\.k_proj$",
      "re:.*\\.self_attn\\.v_proj$",
      "re:.*\\.self_attn\\.o_proj$",
      "re:.*\\.self_attn\\.g_proj$",
      "re:.*\\.mlp\\.gate$",
      "re:.*\\.mlp\\.shared_expert\\.gate_proj$",
      "re:.*\\.mlp\\.shared_expert\\.up_proj$",
      "re:.*\\.mlp\\.shared_expert\\.down_proj$"
    ],
    "kv_cache_scheme": {
      "actorder": null,
      "block_structure": null,
      "dynamic": false,
      "group_size": null,
      "num_bits": 8,
      "observer": "minmax",
      "observer_kwargs": {},
      "scale_dtype": null,
      "strategy": "tensor",
      "symmetric": true,
      "type": "float",
      "zp_dtype": null
    },
    "quant_method": "compressed-tensors",
    "quantization_status": "compressed",
    "sparsity_config": {},
    "transform_config": {
      "config_groups": {
        "R1": {
          "apply": [
            {
              "ignore": [],
              "inverse": false,
              "location": "weight_output",
              "targets": [
                "re:.*embed_tokens$",
                "re:.*o_proj$",
                "re:.*down_proj$"
              ]
            },
            {
              "ignore": [],
              "inverse": true,
              "location": "weight_input",
              "targets": [
                "re:.*q_proj$",
                "re:.*k_proj$",
                "re:.*v_proj$",
                "re:.*gate_proj$",
                "re:.*up_proj$",
                "re:.*mlp.gate$",
                "re:.*g_proj$",
                "re:.*lm_head$"
              ]
            }
          ],
          "head_dim": 128,
          "precision": "torch.float64",
          "randomize": false,
          "requires_grad": false,
          "type": "hadamard"
        }
      }
    },
    "version": "0.14.1.dev11+gf2ee47b"
  }
}