Skip to content

Commit 1d192dd

Browse files
committed
Merge branch 'main' of github.com:painebenjamin/diffusers
2 parents 66c2ab4 + d49cbc9 commit 1d192dd

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/diffusers/pipelines/pag/pipeline_pag_sd_3_img2img.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -952,9 +952,6 @@ def __call__(
952952
generator,
953953
)
954954

955-
num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
956-
self._num_timesteps = len(timesteps)
957-
958955
if self.do_perturbed_attention_guidance:
959956
original_attn_proc = self.transformer.attn_processors
960957
self._set_pag_attn_processor(
@@ -963,6 +960,8 @@ def __call__(
963960
)
964961

965962
# 6. Denoising loop
963+
num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
964+
self._num_timesteps = len(timesteps)
966965
with self.progress_bar(total=num_inference_steps) as progress_bar:
967966
for i, t in enumerate(timesteps):
968967
if self.interrupt:

0 commit comments

Comments
 (0)