diff --git a/api/onnx_web/diffusers/pipelines/controlnet.py b/api/onnx_web/diffusers/pipelines/controlnet.py index 684b0c5e..bcee67bf 100644 --- a/api/onnx_web/diffusers/pipelines/controlnet.py +++ b/api/onnx_web/diffusers/pipelines/controlnet.py @@ -13,7 +13,7 @@ import numpy as np import PIL import torch from diffusers.configuration_utils import FrozenDict -from diffusers.pipeline_utils import DiffusionPipeline +from diffusers.pipelines.pipeline_utils import DiffusionPipeline from diffusers.pipelines.onnx_utils import ORT_TO_NP_TYPE, OnnxRuntimeModel from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput from diffusers.schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler diff --git a/api/onnx_web/diffusers/pipelines/panorama.py b/api/onnx_web/diffusers/pipelines/panorama.py index 78151f12..22473a51 100644 --- a/api/onnx_web/diffusers/pipelines/panorama.py +++ b/api/onnx_web/diffusers/pipelines/panorama.py @@ -19,7 +19,7 @@ import numpy as np import PIL import torch from diffusers.configuration_utils import FrozenDict -from diffusers.pipeline_utils import DiffusionPipeline +from diffusers.pipelines.pipeline_utils import DiffusionPipeline from diffusers.pipelines.onnx_utils import ORT_TO_NP_TYPE, OnnxRuntimeModel from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput from diffusers.schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler diff --git a/api/onnx_web/diffusers/pipelines/pix2pix.py b/api/onnx_web/diffusers/pipelines/pix2pix.py index ffcd7e9b..2689fa4f 100644 --- a/api/onnx_web/diffusers/pipelines/pix2pix.py +++ b/api/onnx_web/diffusers/pipelines/pix2pix.py @@ -32,7 +32,7 @@ except ImportError: } from diffusers import OnnxRuntimeModel -from diffusers.pipeline_utils import DiffusionPipeline +from diffusers.pipelines.pipeline_utils import DiffusionPipeline from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput from diffusers.schedulers import ( DDIMScheduler, diff --git a/api/onnx_web/diffusers/pipelines/upscale.py b/api/onnx_web/diffusers/pipelines/upscale.py index deb2c8c3..db961091 100644 --- a/api/onnx_web/diffusers/pipelines/upscale.py +++ b/api/onnx_web/diffusers/pipelines/upscale.py @@ -11,7 +11,7 @@ from typing import Any, Callable, List, Optional, Union import numpy as np import PIL import torch -from diffusers.pipeline_utils import ImagePipelineOutput +from diffusers.pipelines.pipeline_utils import ImagePipelineOutput from diffusers.pipelines.onnx_utils import ORT_TO_NP_TYPE, OnnxRuntimeModel from diffusers.pipelines.stable_diffusion import StableDiffusionUpscalePipeline from diffusers.schedulers import DDPMScheduler diff --git a/api/pyproject.toml b/api/pyproject.toml index efed7be3..f0c3b689 100644 --- a/api/pyproject.toml +++ b/api/pyproject.toml @@ -41,9 +41,9 @@ module = [ "diffusers.pipelines.latent_diffusion.pipeline_latent_diffusion", "diffusers.pipelines.onnx_utils", "diffusers.pipelines.paint_by_example", + "diffusers.pipelines.pipeline_utils", "diffusers.pipelines.stable_diffusion", "diffusers.pipelines.stable_diffusion.convert_from_ckpt", - "diffusers.pipeline_utils", "diffusers.schedulers", "diffusers.utils.logging", "facexlib.utils",