add args check

inisis · May 16, 2024 · 8c35f63 · 8c35f63
1 parent 73021c4
commit 8c35f63
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/onnxslim/cli/_main.py b/onnxslim/cli/_main.py
@@ -265,6 +265,9 @@ def main():
     if not args.inspect and args.dump_to_disk:
         parser.error("dump_to_disk can only be used with --inspect")
 
+    if args.save_as_external_data and args.output_model:
+        parser.error("--save_as_external_data can only be used for single model")
+
     slim(
         args.input_model,
         args.output_model,