From cb90fd69b4802416b1d5f5e4b3c87e0fcc522d3f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 19 Jul 2022 10:34:52 +0000 Subject: [PATCH] upload code --- src/diffusers/schedulers/scheduling_ddpm.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/schedulers/scheduling_ddpm.py b/src/diffusers/schedulers/scheduling_ddpm.py index 1ef238d362..d5e3264038 100644 --- a/src/diffusers/schedulers/scheduling_ddpm.py +++ b/src/diffusers/schedulers/scheduling_ddpm.py @@ -94,6 +94,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): self.set_format(tensor_format=tensor_format) def set_timesteps(self, num_inference_steps): + num_inference_steps = min(self.config.num_train_timesteps, num_inference_steps) self.num_inference_steps = num_inference_steps self.timesteps = np.arange( 0, self.config.num_train_timesteps, self.config.num_train_timesteps // self.num_inference_steps