Add files via upload

rxfxt · Feb 22, 2023 · 9b31567 · 9b31567
1 parent 63d7a20
commit 9b31567
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/onnx_export.py b/onnx_export.py
@@ -1,6 +1,5 @@
-from modules.diff.diffusion_V2 import GaussianDiffusion
 import utils
-from infer_tools.infer_tool import Svc
+from infer_tools.infer_tool import SvcOnnx
 
 
 
@@ -10,7 +9,7 @@
     config_path = f'./checkpoints/{project_name}/config.yaml'
     hubert_gpu = False
 
-    model = Svc(project_name, config_path, hubert_gpu, model_path, onnx=False)
+    model = SvcOnnx(project_name, config_path, hubert_gpu, model_path)
     model = model.model
     model.cpu()
     model.fs2.cpu()