Unverified Commit fb22be58 authored by Mengqing Cao's avatar Mengqing Cao Committed by GitHub
Browse files

[moe][quant] add weight name case for offset (#15515)


Signed-off-by: default avatarMengqing Cao <cmq0113@163.com>
parent 7f301dd8
......@@ -699,8 +699,9 @@ class FusedMoE(torch.nn.Module):
tp_rank=self.tp_rank)
return
# Case weight scales and zero_points
if ("scale" in weight_name or "zero" in weight_name):
# Case weight scales, zero_points and offset
if ("scale" in weight_name or "zero" in weight_name
or "offset" in weight_name):
# load the weight scales and zp based on the quantization scheme
# supported weight scales/zp can be found in
# FusedMoeWeightScaleSupported
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment