From c148f4331f88ee8b4c9dae37cd3b8d61dee46010 Mon Sep 17 00:00:00 2001
From: Ryan Y <101505765+KeepNoob@users.noreply.github.com>
Date: Wed, 20 Mar 2024 16:47:05 +0800
Subject: [PATCH] Update pipeline_ddpm.py

Add scale_model_input function call before self.scheduler.step in DDPM pipeline to ensure the need of scaling the denoising model input
---
 src/diffusers/pipelines/ddpm/pipeline_ddpm.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/diffusers/pipelines/ddpm/pipeline_ddpm.py b/src/diffusers/pipelines/ddpm/pipeline_ddpm.py
index 093a3cdfe512..3112ef7d5b5d 100644
--- a/src/diffusers/pipelines/ddpm/pipeline_ddpm.py
+++ b/src/diffusers/pipelines/ddpm/pipeline_ddpm.py
@@ -110,6 +110,9 @@ def __call__(
         self.scheduler.set_timesteps(num_inference_steps)
 
         for t in self.progress_bar(self.scheduler.timesteps):
+            # scale_model_input
+            image = self.scheduler.scale_model_input(image, t)
+            
             # 1. predict noise model_output
             model_output = self.unet(image, t).sample