From df1504d9b095de6d0cdfaa7fcf60ed7e70649c43 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 10 Mar 2023 18:50:01 -0600 Subject: [PATCH] use fake VAE config --- .../diffusers/pipeline_onnx_stable_diffusion_upscale.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 d5018274..1977ea76 100644 --- a/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py +++ b/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py @@ -78,9 +78,9 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): max_noise_level: int = 350, ): if hasattr(vae, "config") == False: - setattr(vae, "config", { - "scaling_factor": 0.08333, - }) + config = {} + setattr(config, "scaling_factor", 0.08333) + setattr(vae, "config", config) super().__init__(vae, text_encoder, tokenizer, unet, low_res_scheduler, scheduler, max_noise_level)