apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use fake VAE config

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

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

@ -78,9 +78,9 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline):
max_noise_level: int = 350, max_noise_level: int = 350,
): ):
if hasattr(vae, "config") == False: if hasattr(vae, "config") == False:
setattr(vae, "config", { config = {}
"scaling_factor": 0.08333, setattr(config, "scaling_factor", 0.08333)
}) setattr(vae, "config", config)
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)