small changes

Author: xadupre

.gitignore

@@ -22,6 +22,7 @@
 *.svg
 *.onnx.stats
 *.stats
+*.zip
 CodeLlama*
 _tools/benchenv**
 _tools/repos**

_scripts/export_qwen25_vl_visual.py

@@ -19,6 +19,11 @@
 
     python export_qwen25_vl_visual.py -m Qwen/Qwen2.5-VL-7B-Instruct --device cpu --dtype float32 --exporter onnx-dynamo --pretrained --second-input
 
+Cheat sheet for tar commmands. To make a tar:
+``tar -czvf model.tar.gz model.onnx model.data``
+And to untar:
+``tar -xzvf model.tar.gz``.
+
 Attention
 +++++++++
 
@@ -32,6 +37,7 @@
 """
 
 import os
+import subprocess
 import sys
 import time
 from argparse import ArgumentParser, BooleanOptionalAction
@@ -60,6 +66,7 @@ def main(
     exporter: str = "onnx-dynamo",
     pretrained: bool = True,
     second_input: bool = True,
+    zip: bool = False,
 ):
     print("-- import torch")
     import torch
@@ -158,9 +165,10 @@ def _config_reduction(config, task):
     prefix = simplify_model_id_for_a_filename(model_id)
     if "QWEN25ATTENTION" in os.environ:
         prefix = f"{prefix}.{os.environ['QWEN25ATTENTION']}"
-    filename = f"model.{prefix}.visual.{device}.{dtype}.{exporter}.onnx"
+    basename = f"model.{prefix}.visual.{device}.{dtype}.{exporter}"
+    filename = f"{basename}.onnx"
     print(f"-- export in {filename!r}")
-    stat_file = filename.replace(".onnx", ".stats")
+    stat_file = f"{basename}.stats"
     begin = time.perf_counter()
 
     if exporter == "onnx-dynamo" and device == "cuda" and "QWEN25ATTENTION" not in os.environ:
@@ -229,6 +237,18 @@ def fprint(s):
             diff = max_diff(expected_big, big[0], hist=[0.1])
             fprint(f"-- discrepancies={diff}")
 
+    if zip:
+        tar_file_name = f"{basename}.zip"
+        print()
+        print(f"-- make file {tar_file_name!r}")
+        cmd = ["zip", "-v", "-1", tar_file_name]
+        for name in [filename, f"{filename}.data"]:
+            print(f"-- add {name!r}")
+            cmd.append(name)
+        print(f"-- cmd: {' '.join(cmd)}")
+        subprocess.run(cmd, check=True)
+        print("-- done.")
+
 
 def get_parser() -> ArgumentParser:
     parser = ArgumentParser(
@@ -260,6 +280,12 @@ def get_parser() -> ArgumentParser:
         help="check discrepancies with other inputs",
         action=BooleanOptionalAction,
     )
+    parser.add_argument(
+        "--zip",
+        default=False,
+        help="Creates a file .zip with onnx file and data file.",
+        action=BooleanOptionalAction,
+    )
     return parser
 
 
@@ -273,4 +299,5 @@ def get_parser() -> ArgumentParser:
         exporter=args.exporter,
         pretrained=args.pretrained,
         second_input=args.second_input,
+        zip=args.zip,
     )