From 9ffe266384794df1147f074b9eddf46057bddde6 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 24 Dec 2023 05:32:30 -0600
Subject: [PATCH] fix(api): unload UNet before CNet conversion

---
 api/onnx_web/convert/diffusion/diffusion.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/api/onnx_web/convert/diffusion/diffusion.py b/api/onnx_web/convert/diffusion/diffusion.py
index abc52578..303bb8e6 100644
--- 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(