diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusers/load.py similarity index 100% rename from api/onnx_web/diffusion/load.py rename to api/onnx_web/diffusers/load.py diff --git a/api/onnx_web/diffusion/lpw_stable_diffusion_onnx.py b/api/onnx_web/diffusers/lpw_stable_diffusion_onnx.py similarity index 100% rename from api/onnx_web/diffusion/lpw_stable_diffusion_onnx.py rename to api/onnx_web/diffusers/lpw_stable_diffusion_onnx.py diff --git a/api/onnx_web/diffusion/no-init.txt b/api/onnx_web/diffusers/no-init.txt similarity index 100% rename from api/onnx_web/diffusion/no-init.txt rename to api/onnx_web/diffusers/no-init.txt diff --git a/api/onnx_web/diffusion/pipeline_onnx_stable_diffusion_upscale.py b/api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py similarity index 100% rename from api/onnx_web/diffusion/pipeline_onnx_stable_diffusion_upscale.py rename to api/onnx_web/diffusers/pipeline_onnx_stable_diffusion_upscale.py diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusers/run.py similarity index 100% rename from api/onnx_web/diffusion/run.py rename to api/onnx_web/diffusers/run.py diff --git a/api/onnx_web/diffusion/stub_scheduler.py b/api/onnx_web/diffusers/stub_scheduler.py similarity index 100% rename from api/onnx_web/diffusion/stub_scheduler.py rename to api/onnx_web/diffusers/stub_scheduler.py diff --git a/api/onnx_web/transformers.py b/api/onnx_web/transformers/run.py similarity index 89% rename from api/onnx_web/transformers.py rename to api/onnx_web/transformers/run.py index 299d3854..25de1071 100644 --- a/api/onnx_web/transformers.py +++ b/api/onnx_web/transformers/run.py @@ -1,8 +1,8 @@ from logging import getLogger -from .params import ImageParams, Size -from .server import ServerContext -from .worker import WorkerContext +from ..params import ImageParams, Size +from ..server import ServerContext +from ..worker import WorkerContext logger = getLogger(__name__)