Skip to content

Commit 9260ca7

Browse files
authored
fix w_group_size KeyError in DeepSeek per-tensor quant (Tencent#50)
1 parent 0c628aa commit 9260ca7

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

angelslim/models/llm/deepseek.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def get_qdq_module(self, layer, name):
163163
quant_algo=self.quant_config.quant_algo,
164164
weight=weight,
165165
weight_scale=weight_scale,
166-
group_size=self.quant_config.quant_algo_info["w_group_size"],
166+
group_size=self.quant_config.quant_algo_info.get("w_group_size", 128),
167167
bias=layer.bias,
168168
input_scale=act_scale,
169169
)

0 commit comments

Comments
 (0)