From cd11e8f3ab9f1a7d78b1e94b6a9f25b4c311e660 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 10 Mar 2023 18:57:01 -0600 Subject: [PATCH] use a proper fake config class --- .../pipeline_onnx_stable_diffusion_upscale.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 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 index 1977ea76..470fafe5 100644 --- a/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py +++ b/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py @@ -65,6 +65,12 @@ def preprocess(image): return image +class FakeConfig(): + scaling_factor: float + + def __init__(self) -> None: + self.scaling_factor = 0.08333 + class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): def __init__( @@ -78,9 +84,7 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline): max_noise_level: int = 350, ): if hasattr(vae, "config") == False: - config = {} - setattr(config, "scaling_factor", 0.08333) - setattr(vae, "config", config) + setattr(vae, "config", FakeConfig()) super().__init__(vae, text_encoder, tokenizer, unet, low_res_scheduler, scheduler, max_noise_level)