alexwww94 commited on
Commit
bd32c15
1 Parent(s): 9f2485d

Update config.json

Browse files
Files changed (1) hide show
  1. 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
  }