fix(api): GC during SDXL conversion

ssube · Dec 24, 2023 · a5a60d3 · a5a60d3
1 parent f6e6e31
commit a5a60d3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/api/onnx_web/convert/diffusion/diffusion_xl.py b/api/onnx_web/convert/diffusion/diffusion_xl.py
@@ -10,6 +10,7 @@
 from optimum.exporters.onnx import main_export
 
 from ...constants import ONNX_MODEL, ONNX_WEIGHTS
+from ...utils import run_gc
 from ..client import fetch_model
 from ..utils import RESOLVE_FORMATS, ConversionContext, check_ext
 
@@ -79,6 +80,10 @@ def convert_diffusion_diffusers_xl(
         logger.debug("exporting torch model for %s: %s", source, temp_path)
         pipeline.save_pretrained(temp_path)
 
+    # GC temporary pipeline
+    del pipeline
+    run_gc()
+
     # directory -> onnx using optimum exporters
     main_export(
         temp_path,