Add files via upload

BEHPOZL · Feb 22, 2023 · ef0c1a0 · ef0c1a0
1 parent 8146a9c
commit ef0c1a0
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/onnx_export.py 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)
+