Update torch_to_onnx_exporter.py

neuralmagic · Jun 12, 2023 · 6dbf554 · 6dbf554
1 parent ec62ff4
commit 6dbf554
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparseml/pytorch/torch_to_onnx_exporter.py b/src/sparseml/pytorch/torch_to_onnx_exporter.py
@@ -148,7 +148,7 @@ def post_validate(self, model: onnx.ModelProto) -> onnx.ModelProto:
         return model
 
     def transform(
-        self, module: torch.nn.Module, file_name: str = "model.onnx"
+        self, module: torch.nn.Module, file_name: str
     ) -> onnx.ModelProto:
         file_path = os.path.join(os.getcwd(), file_name)