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__(