apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correct VAE config check

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

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

@ -77,7 +77,7 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline):
scheduler: Any, scheduler: Any,
max_noise_level: int = 350, max_noise_level: int = 350,
): ):
if getattr(vae, "config") is None: if hasattr(vae, "config") == False:
setattr(vae, "config", { setattr(vae, "config", {
"scaling_factor": 0.08333, "scaling_factor": 0.08333,
}) })