add version arg

inisis · Aug 17, 2024 · 2afac54 · 2afac54
1 parent a276c71
commit 2afac54
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnxslim/argparser.py b/onnxslim/argparser.py
@@ -109,6 +109,7 @@ def _add_arguments(self):
         # Add positional arguments separately for ModelArguments
         self.parser.add_argument("input_model", help="input onnx model")
         self.parser.add_argument("output_model", nargs="?", default=None, help="output onnx model")
+        self.parser.add_argument("-v", "--version", action="version", version=onnxslim.__version__)
 
     def parse_args_into_dataclasses(self):
         args = self.parser.parse_args()