From 30bf95d49362c3ad4f956a4eac9488a2bd6ae5fb Mon Sep 17 00:00:00 2001
From: unanmed <1319491857@qq.com>
Date: Mon, 28 Apr 2025 14:26:54 +0800
Subject: [PATCH] feat: save model as deployment

---
 ginka/save.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 ginka/save.py

diff --git a/ginka/save.py b/ginka/save.py
new file mode 100644
index 0000000..e671f2f
--- /dev/null
+++ b/ginka/save.py
@@ -0,0 +1,16 @@
+import argparse
+import torch
+
+def to_deployment(path: str, output: str):
+    state = torch.load(path)
+    torch.save({
+        "model_state": state["model_state"]
+    }, output)
+    
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--input", type=str, default="result/ginka.pth")
+    parser.add_argument("--output", type=str, default="result/ginka_deploy.pth")
+    args = parser.parse_args()
+    to_deployment(args.input, args.output)
+    
\ No newline at end of file