We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e7d09bb commit de38272Copy full SHA for de38272
modelopt/onnx/quantization/quantize.py
@@ -141,7 +141,7 @@ def _preprocess_onnx(
141
raise e
142
143
try:
144
- model_simp = onnxslim.slim(onnx_model)
+ model_simp = onnxslim.slim(onnx_model, skip_fusion_patterns=["FusionGemm"])
145
if model_simp:
146
onnx_model = model_simp
147
onnx_path = os.path.join(output_dir, f"{model_name}_simp.onnx")
0 commit comments