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,
|
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__(
|
||||||
|
|
Loading…
Reference in New Issue