Skip to content

Commit a4e80d8

Browse files
committed
fix t_index
1 parent 0c8e83f commit a4e80d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -311,7 +311,7 @@ def __call__(
311311
if isinstance(self.scheduler, LMSDiscreteScheduler):
312312
latents = self.scheduler.step(noise_pred, t_index, latents, **extra_step_kwargs).prev_sample
313313
# masking
314-
init_latents_proper = self.scheduler.add_noise(init_latents_orig, noise, torch.tensor(t_index))
314+
init_latents_proper = self.scheduler.add_noise(init_latents_orig, noise, torch.LongTensor([t_index]))
315315
else:
316316
latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs).prev_sample
317317
# masking

0 commit comments

Comments
 (0)