From 00b1bec7a9d992c043a329316ea6d12643d1b999 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 27 May 2023 12:31:08 -0500
Subject: [PATCH] fix(api): skip loading custom VAE after torch extraction
 (#379)

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

diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py
index 14fc4dc3..8fcd5562 100644
--- a/api/onnx_web/convert/diffusion/diffusers.py
+++ b/api/onnx_web/convert/diffusion/diffusers.py
@@ -344,6 +344,9 @@ def convert_diffusion_diffusers(
                 torch_source,
                 torch_dtype=dtype,
             ).to(device)
+
+            # VAE replacement already happened during extraction, skip
+            replace_vae = None
         else:
             logger.debug("loading pipeline from SD checkpoint: %s", source)
             pipeline = download_from_original_stable_diffusion_ckpt(