diff --git a/onnx_export.py b/onnx_export.py
new file mode 100644
index 0000000..e668479
--- /dev/null
+++ b/onnx_export.py
@@ -0,0 +1,19 @@
+from modules.diff.diffusion_V2 import GaussianDiffusion
+import utils
+from infer_tools.infer_tool import Svc
+
+
+
+if __name__ == "__main__":
+    project_name = "Yua"
+    model_path = f'./checkpoints/{project_name}/model.ckpt'
+    config_path = f'./checkpoints/{project_name}/config.yaml'
+    hubert_gpu = False
+
+    model = Svc(project_name, config_path, hubert_gpu, model_path, onnx=False)
+    model = model.model
+    model.cpu()
+    model.fs2.cpu()
+
+    model.OnnxExport(project_name)
+