apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

increment scheduler index after view loop

This commit is contained in:
Sean Sube 2024-01-14 21:46:47 -06:00
parent d7af93c34a
commit bb1b3095c8
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
api/onnx_web/diffusers/pipelines/panorama_xl.py
View File

@ -411,6 +411,8 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order num_warmup_steps = len(timesteps) - num_inference_steps * self.scheduler.order
for i, t in enumerate(self.progress_bar(timesteps)): for i, t in enumerate(self.progress_bar(timesteps)):
last = i == (len(timesteps) - 1) last = i == (len(timesteps) - 1)
next_step_index = None
count.fill(0) count.fill(0)
value.fill(0) value.fill(0)
@ -471,10 +473,11 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
# reset the scheduler's internal timestep # reset the scheduler's internal timestep
if prev_step_index is not None: if prev_step_index is not None:
logger.debug( logger.debug(
"resetting scheduler internal step index from %s to %s", "rewinding scheduler internal step index from %s to %s",
self.scheduler._step_index, self.scheduler._step_index,
prev_step_index, prev_step_index,
) )
next_step_index = self.scheduler._step_index
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 value[:, :, h_start:h_end, w_start:w_end] += latents_view_denoised
@ -572,6 +575,7 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
self.scheduler._step_index, self.scheduler._step_index,
prev_step_index, prev_step_index,
) )
next_step_index = self.scheduler._step_index
self.scheduler._step_index = prev_step_index self.scheduler._step_index = prev_step_index
if feather[0] > 0.0: if feather[0] > 0.0:
@ -602,6 +606,15 @@ class StableDiffusionXLPanoramaPipelineMixin(StableDiffusionXLImg2ImgPipelineMix
latents = np.where(count > 0, value / count, value) latents = np.where(count > 0, value / count, value)
latents = repair_nan(latents) latents = repair_nan(latents)
# update the scheduler's internal timestep, if set
if next_step_index is not None:
logger.debug(
"forwarding scheduler internal step index from %s to %s",
self.scheduler._step_index,
next_step_index,
)
self.scheduler._step_index = next_step_index
# call the callback, if provided # call the callback, if provided
if i == len(timesteps) - 1 or ( if i == len(timesteps) - 1 or (
(i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0 (i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0