apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): update deprecated import from diffusers

This commit is contained in:
Sean Sube 2023-11-23 11:34:18 -06:00
parent f00bfe9bd0
commit 3ffb30f121
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/diffusers/pipelines/panorama_xl.py
View File

@ -10,7 +10,8 @@ from optimum.onnxruntime.modeling_diffusion import ORTStableDiffusionXLPipelineB
from optimum.pipelines.diffusers.pipeline_stable_diffusion_xl_img2img import ( from optimum.pipelines.diffusers.pipeline_stable_diffusion_xl_img2img import (
StableDiffusionXLImg2ImgPipelineMixin, 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 from onnx_web.chain.tile import make_tile_mask
@ -728,7 +729,8 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
) )
# 3. Preprocess image # 3. Preprocess image
image = preprocess(image) processor = VaeImageProcessor()
image = processor.preprocess(image)
# 4. Prepare timesteps # 4. Prepare timesteps
self.scheduler.set_timesteps(num_inference_steps) self.scheduler.set_timesteps(num_inference_steps)