Update config.json
Browse files- config.json +7 -1
config.json
CHANGED
@@ -67,6 +67,12 @@
|
|
67 |
"model_name_or_path": null,
|
68 |
"model_file_base_name": null,
|
69 |
"quant_method": "gptq",
|
70 |
-
"checkpoint_format": "gptq"
|
|
|
|
|
|
|
|
|
|
|
|
|
71 |
}
|
72 |
}
|
|
|
67 |
"model_name_or_path": null,
|
68 |
"model_file_base_name": null,
|
69 |
"quant_method": "gptq",
|
70 |
+
"checkpoint_format": "gptq",
|
71 |
+
"block_name_to_quantize": "",
|
72 |
+
"modules_in_block_to_quantize": [
|
73 |
+
["self_attention.query_key_value", "self_attention.dense", "mlp.dense_h_to_4h", "mlp.dense_4h_to_h"],
|
74 |
+
["attention.query_key_value", "attention.dense", "mlp.fc1", "mlp.fc2"],
|
75 |
+
["linear_proj", "dense_h_to_4h", "gate_proj", "dense_4h_to_h"],
|
76 |
+
]
|
77 |
}
|
78 |
}
|