fix(api): skip loading custom VAE after torch extraction (#379)

ssube · May 27, 2023 · 00b1bec · 00b1bec
1 parent 3adafea
commit 00b1bec
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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(