Skip to content

Commit 7240318

Browse files
authored
Fix the order of casts for onnx inpainting (#1338)
1 parent aa2ce41 commit 7240318

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -409,8 +409,8 @@ def __call__(
409409
latent_model_input = np.concatenate([latents] * 2) if do_classifier_free_guidance else latents
410410
# concat latents, mask, masked_image_latnets in the channel dimension
411411
latent_model_input = self.scheduler.scale_model_input(torch.from_numpy(latent_model_input), t)
412-
latent_model_input = np.concatenate([latent_model_input, mask, masked_image_latents], axis=1)
413412
latent_model_input = latent_model_input.cpu().numpy()
413+
latent_model_input = np.concatenate([latent_model_input, mask, masked_image_latents], axis=1)
414414

415415
# predict the noise residual
416416
timestep = np.array([t], dtype=timestep_dtype)

0 commit comments

Comments
 (0)