diff --git a/api/onnx_web/diffusers/pipelines/panorama.py b/api/onnx_web/diffusers/pipelines/panorama.py index c3c518fe..cd9a32d0 100644 --- a/api/onnx_web/diffusers/pipelines/panorama.py +++ b/api/onnx_web/diffusers/pipelines/panorama.py @@ -628,6 +628,11 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline): # reset the scheduler's internal timestep if prev_step_index is not None: + logger.debug( + "resetting scheduler internal step index from %s to %s", + self.scheduler._step_index, + prev_step_index, + ) self.scheduler._step_index = prev_step_index value[:, :, h_start:h_end, w_start:w_end] += latents_view_denoised @@ -711,6 +716,11 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline): # reset the scheduler's internal timestep if prev_step_index is not None: + logger.debug( + "resetting scheduler internal step index from %s to %s", + self.scheduler._step_index, + prev_step_index, + ) self.scheduler._step_index = prev_step_index if feather[0] > 0.0: @@ -1061,6 +1071,11 @@ class OnnxStableDiffusionPanoramaPipeline(DiffusionPipeline): # reset the scheduler's internal timestep if prev_step_index is not None: + logger.debug( + "resetting scheduler internal step index from %s to %s", + self.scheduler._step_index, + prev_step_index, + ) self.scheduler._step_index = prev_step_index value[:, :, h_start:h_end, w_start:w_end] += latents_view_denoised diff --git a/api/onnx_web/diffusers/pipelines/panorama_xl.py b/api/onnx_web/diffusers/pipelines/panorama_xl.py index 70f3be28..e15915b3 100644 --- a/api/onnx_web/diffusers/pipelines/panorama_xl.py +++ b/api/onnx_web/diffusers/pipelines/panorama_xl.py @@ -470,6 +470,11 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix # reset the scheduler's internal timestep if prev_step_index is not None: + logger.debug( + "resetting scheduler internal step index from %s to %s", + self.scheduler._step_index, + prev_step_index, + ) self.scheduler._step_index = prev_step_index value[:, :, h_start:h_end, w_start:w_end] += latents_view_denoised @@ -562,6 +567,11 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix # reset the scheduler's internal timestep if prev_step_index is not None: + logger.debug( + "resetting scheduler internal step index from %s to %s", + self.scheduler._step_index, + prev_step_index, + ) self.scheduler._step_index = prev_step_index if feather[0] > 0.0: @@ -912,6 +922,11 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix # reset the scheduler's internal timestep if prev_step_index is not None: + logger.debug( + "resetting scheduler internal step index from %s to %s", + self.scheduler._step_index, + prev_step_index, + ) self.scheduler._step_index = prev_step_index value[:, :, h_start:h_end, w_start:w_end] += latents_view_denoised diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py index 0477a7f5..6606be69 100644 --- a/api/onnx_web/diffusers/run.py +++ b/api/onnx_web/diffusers/run.py @@ -5,7 +5,6 @@ from typing import Any, List, Optional from PIL import Image, ImageOps from ..chain import ( - BlendDenoiseFastNLMeansStage, BlendImg2ImgStage, BlendMaskStage, ChainPipeline, @@ -130,12 +129,6 @@ def run_txt2img_pipeline( # apply upscaling and correction, before highres highres_size = get_highres_tile(server, params, highres, tile_size) - if params.is_panorama(): - chain.stage( - BlendDenoiseFastNLMeansStage(), - StageParams(tile_size=highres_size), - ) - first_upscale, after_upscale = split_upscale(upscale) if first_upscale: stage_upscale_correction(