From 2033cbb601f1553bb749bf0f1e84c0d579e8ad11 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 29 Jan 2023 11:39:02 -0600 Subject: [PATCH] fix tile latent axis --- api/onnx_web/diffusion/load.py | 5 ++--- .../pipeline_onnx_stable_diffusion_upscale.py | 0 api/onnx_web/onnx/__init__.py | 3 --- 3 files changed, 2 insertions(+), 6 deletions(-) rename api/onnx_web/{onnx => diffusion}/pipeline_onnx_stable_diffusion_upscale.py (100%) diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py index 81e3a2fc..2f8a7129 100644 --- a/api/onnx_web/diffusion/load.py +++ b/api/onnx_web/diffusion/load.py @@ -33,14 +33,13 @@ def get_latents_from_seed(seed: int, size: Size) -> np.ndarray: def get_tile_latents(full_latents: np.ndarray, dims: Tuple[int, int, int]) -> np.ndarray: x, y, tile = dims + t = tile // 8 x = x // 8 y = y // 8 - - t = tile // 8 xt = x + t yt = y + t - return full_latents[:,:,x:xt,y:yt] + return full_latents[:,:,y:yt,x:xt] def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, scheduler: Any, device: Optional[str] = None): diff --git a/api/onnx_web/onnx/pipeline_onnx_stable_diffusion_upscale.py b/api/onnx_web/diffusion/pipeline_onnx_stable_diffusion_upscale.py similarity index 100% rename from api/onnx_web/onnx/pipeline_onnx_stable_diffusion_upscale.py rename to api/onnx_web/diffusion/pipeline_onnx_stable_diffusion_upscale.py diff --git a/api/onnx_web/onnx/__init__.py b/api/onnx_web/onnx/__init__.py index 693abcc5..609d3473 100644 --- a/api/onnx_web/onnx/__init__.py +++ b/api/onnx_web/onnx/__init__.py @@ -1,7 +1,4 @@ from .onnx_net import ( OnnxImage, OnnxNet, -) -from .pipeline_onnx_stable_diffusion_upscale import ( - OnnxStableDiffusionUpscalePipeline, ) \ No newline at end of file