From e960ee1d8510b2378292ab80d6851bb2dc541e25 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 10 Mar 2023 18:42:11 -0600 Subject: [PATCH] fix(api): patch VAE for SD upscaling on older versions of diffusers (#234) --- .../diffusers/pipeline_onnx_stable_diffusion_upscale.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py b/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py index de60a80c..8bf8bde2 100644 --- a/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py +++ b/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py @@ -77,6 +77,11 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): scheduler: Any, max_noise_level: int = 350, ): + if getattr(vae, "config") is None: + setattr(vae, "config", { + "scaling_factor": 0.08333, + }) + super().__init__(vae, text_encoder, tokenizer, unet, low_res_scheduler, scheduler, max_noise_level) def __call__(