fix(api): patch VAE for SD upscaling on older versions of diffusers (#…

…234)
ssube · Mar 11, 2023 · e960ee1 · e960ee1
1 parent 6bdaa4a
commit e960ee1
Showing 1 changed file with 5 additions and 0 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
@@ -77,6 +77,11 @@ def __init__(
         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__(