Quality fixes

neuralmagic · Oct 3, 2023 · 8d63a8a · 8d63a8a
1 parent 4855898
commit 8d63a8a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/sparseml/exporters/onnx_to_deepsparse.py b/src/sparseml/exporters/onnx_to_deepsparse.py
@@ -80,7 +80,9 @@ def __init__(
             sparseml_transforms.PropagateDequantThroughSplit(),
         ]
         if use_qlinear_matmul:
-            transforms.append(sparseml_transforms.MatMulToQLinearMatMul(),)
+            transforms.append(
+                sparseml_transforms.MatMulToQLinearMatMul(),
+            )
 
         transforms.extend(
             [