Skip to content

Commit de38272

Browse files
committed
fix: disable Gemm Add fusion
Signed-off-by: inisis <[email protected]>
1 parent e7d09bb commit de38272

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/onnx/quantization/quantize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ def _preprocess_onnx(
141141
raise e
142142

143143
try:
144-
model_simp = onnxslim.slim(onnx_model)
144+
model_simp = onnxslim.slim(onnx_model, skip_fusion_patterns=["FusionGemm"])
145145
if model_simp:
146146
onnx_model = model_simp
147147
onnx_path = os.path.join(output_dir, f"{model_name}_simp.onnx")

0 commit comments

Comments
 (0)