fix(api): update deprecated import from diffusers

ssube · Nov 23, 2023 · 3ffb30f · 3ffb30f
1 parent f00bfe9
commit 3ffb30f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/api/onnx_web/diffusers/pipelines/panorama_xl.py b/api/onnx_web/diffusers/pipelines/panorama_xl.py
@@ -10,7 +10,8 @@
 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 @@ def img2img(
         )
 
         # 3. Preprocess image
-        image = preprocess(image)
+        processor = VaeImageProcessor()
+        image = processor.preprocess(image)
 
         # 4. Prepare timesteps
         self.scheduler.set_timesteps(num_inference_steps)