apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-03-10 18:42:11 -06:00
parent 6bdaa4a4ed
commit e960ee1d85
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py
View File

@ -77,6 +77,11 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline):
scheduler: Any, scheduler: Any,
max_noise_level: int = 350, 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) super().__init__(vae, text_encoder, tokenizer, unet, low_res_scheduler, scheduler, max_noise_level)
def __call__( def __call__(