From 3e69e241f794e81d1cfab8ae29695ab384fefaea Mon Sep 17 00:00:00 2001 From: Seongbin Lim <58146755+sbinnee@users.noreply.github.com> Date: Tue, 24 Sep 2024 08:28:14 +0900 Subject: [PATCH] Allow DDPMPipeline half precision (#9222) Co-authored-by: YiYi Xu --- src/diffusers/pipelines/ddpm/pipeline_ddpm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/ddpm/pipeline_ddpm.py b/src/diffusers/pipelines/ddpm/pipeline_ddpm.py index 093a3cdfe5..bb03a8d667 100644 --- a/src/diffusers/pipelines/ddpm/pipeline_ddpm.py +++ b/src/diffusers/pipelines/ddpm/pipeline_ddpm.py @@ -101,10 +101,10 @@ class DDPMPipeline(DiffusionPipeline): if self.device.type == "mps": # randn does not work reproducibly on mps - image = randn_tensor(image_shape, generator=generator) + image = randn_tensor(image_shape, generator=generator, dtype=self.unet.dtype) image = image.to(self.device) else: - image = randn_tensor(image_shape, generator=generator, device=self.device) + image = randn_tensor(image_shape, generator=generator, device=self.device, dtype=self.unet.dtype) # set step values self.scheduler.set_timesteps(num_inference_steps)