diff --git a/src/diffusers/pipelines/pndm/pipeline_pndm.py b/src/diffusers/pipelines/pndm/pipeline_pndm.py index 3c3c36f0dc..7b8946529c 100644 --- a/src/diffusers/pipelines/pndm/pipeline_pndm.py +++ b/src/diffusers/pipelines/pndm/pipeline_pndm.py @@ -49,16 +49,6 @@ class PNDMPipeline(DiffusionPipeline): image = self.scheduler.step(model_output, t, image)["prev_sample"] - # for i, t in enumerate(tqdm(self.scheduler.prk_timesteps)): - # model_output = self.unet(image, t)["sample"] - # - # image = self.scheduler.step_prk(model_output, t, image, i=i)["prev_sample"] - # - # for i, t in enumerate(tqdm(self.scheduler.plms_timesteps)): - # model_output = self.unet(image, t)["sample"] - # - # image = self.scheduler.step_plms(model_output, t, image, i=i)["prev_sample"] - image = (image / 2 + 0.5).clamp(0, 1) image = image.cpu().permute(0, 2, 3, 1).numpy() if output_type == "pil":