Skip to content

Commit 560a61d

Browse files
[Bug fix] Make sure correct timesteps are chosen for img2img (huggingface#3128)
Make sure correct timesteps are chosen for img2img
1 parent 7534b51 commit 560a61d

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -503,7 +503,7 @@ def get_timesteps(self, num_inference_steps, strength, device):
503503
init_timestep = min(int(num_inference_steps * strength), num_inference_steps)
504504

505505
t_start = max(num_inference_steps - init_timestep, 0)
506-
timesteps = self.scheduler.timesteps[t_start:]
506+
timesteps = self.scheduler.timesteps[t_start * self.scheduler.order :]
507507

508508
return timesteps, num_inference_steps - t_start
509509

pipelines/stable_diffusion/pipeline_cycle_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -528,7 +528,7 @@ def get_timesteps(self, num_inference_steps, strength, device):
528528
init_timestep = min(int(num_inference_steps * strength), num_inference_steps)
529529

530530
t_start = max(num_inference_steps - init_timestep, 0)
531-
timesteps = self.scheduler.timesteps[t_start:]
531+
timesteps = self.scheduler.timesteps[t_start * self.scheduler.order :]
532532

533533
return timesteps, num_inference_steps - t_start
534534

pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,7 @@ def get_timesteps(self, num_inference_steps, strength, device):
390390
init_timestep = min(int(num_inference_steps * strength), num_inference_steps)
391391

392392
t_start = max(num_inference_steps - init_timestep, 0)
393-
timesteps = self.scheduler.timesteps[t_start:]
393+
timesteps = self.scheduler.timesteps[t_start * self.scheduler.order :]
394394

395395
return timesteps, num_inference_steps - t_start
396396

pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -511,7 +511,7 @@ def get_timesteps(self, num_inference_steps, strength, device):
511511
init_timestep = min(int(num_inference_steps * strength), num_inference_steps)
512512

513513
t_start = max(num_inference_steps - init_timestep, 0)
514-
timesteps = self.scheduler.timesteps[t_start:]
514+
timesteps = self.scheduler.timesteps[t_start * self.scheduler.order :]
515515

516516
return timesteps, num_inference_steps - t_start
517517

pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -507,7 +507,7 @@ def get_timesteps(self, num_inference_steps, strength, device):
507507
init_timestep = min(int(num_inference_steps * strength), num_inference_steps)
508508

509509
t_start = max(num_inference_steps - init_timestep, 0)
510-
timesteps = self.scheduler.timesteps[t_start:]
510+
timesteps = self.scheduler.timesteps[t_start * self.scheduler.order :]
511511

512512
return timesteps, num_inference_steps - t_start
513513

0 commit comments

Comments
 (0)