From 3ffb30f1217aa66a705a382410cc7e3059b2d612 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 23 Nov 2023 11:34:18 -0600 Subject: [PATCH] fix(api): update deprecated import from diffusers --- api/onnx_web/diffusers/pipelines/panorama_xl.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusers/pipelines/panorama_xl.py b/api/onnx_web/diffusers/pipelines/panorama_xl.py index 9d4dfc08..9a4ec773 100644 --- a/api/onnx_web/diffusers/pipelines/panorama_xl.py +++ b/api/onnx_web/diffusers/pipelines/panorama_xl.py @@ -10,7 +10,8 @@ from optimum.onnxruntime.modeling_diffusion import ORTStableDiffusionXLPipelineB from optimum.pipelines.diffusers.pipeline_stable_diffusion_xl_img2img import ( StableDiffusionXLImg2ImgPipelineMixin, ) -from optimum.pipelines.diffusers.pipeline_utils import preprocess, rescale_noise_cfg +from optimum.pipelines.diffusers.pipeline_utils import rescale_noise_cfg +from diffusers.image_processor import VaeImageProcessor from onnx_web.chain.tile import make_tile_mask @@ -728,7 +729,8 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix ) # 3. Preprocess image - image = preprocess(image) + processor = VaeImageProcessor() + image = processor.preprocess(image) # 4. Prepare timesteps self.scheduler.set_timesteps(num_inference_steps)