remove panorama denoise stage
This commit is contained in:
parent
374efca776
commit
d7af93c34a
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue