Merge branch 'main' into patch-45

torchchat/export.py

@@ -522,3 +522,12 @@ def main(args):
                     package=True,
                     metadata=metadata,
                 )
+
+        if output_snapshot_path:
+            output_snapshot_path = str(os.path.abspath(output_snapshot_path))
+            print(f"Exporting model using Snapshot to {output_snapshot_path}")
+            export_snapshot(
+                model_to_snapshot,
+                builder_args.device,
+                output_snapshot_path,
+            )