From 555de5e07d490980929ef3b5c5657363de2952dc Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 10 Mar 2023 18:46:07 -0600 Subject: [PATCH] fix(api): correct VAE config check --- .../diffusers/pipeline_onnx_stable_diffusion_upscale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 8bf8bde2..d5018274 100644 --- a/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py +++ b/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py @@ -77,7 +77,7 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): scheduler: Any, max_noise_level: int = 350, ): - if getattr(vae, "config") is None: + if hasattr(vae, "config") == False: setattr(vae, "config", { "scaling_factor": 0.08333, })