fix(api): patch VAE for SD upscaling on older versions of diffusers (#234)
This commit is contained in:
parent
6bdaa4a4ed
commit
e960ee1d85
|
@ -77,6 +77,11 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline):
|
|||
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__(
|
||||
|
|
Loading…
Reference in New Issue