fix(api): unload UNet before CNet conversion

ssube · Dec 24, 2023 · 9ffe266 · 9ffe266
1 parent 054e5c4
commit 9ffe266
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py
@@ -878,6 +878,9 @@ def convert_diffusion_diffusers_optimum(
             location=ONNX_WEIGHTS,
         )
 
+        del unet
+        run_gc()
+
     if conversion.control:
         logger.debug("converting CNet from pretrained UNet")
         cnet_path = convert_diffusion_diffusers_cnet(