diff --git a/b/src/diffusers/schedulers/scheduling_ddpm.py b/b/src/diffusers/schedulers/scheduling_ddpm.py new file mode 100644 index 000000000000..ab75ed76a822 --- /dev/null +++ b/b/src/diffusers/schedulers/scheduling_ddpm.py @@ -0,0 +1,10 @@ + timesteps = np.array(timesteps, dtype=np.int64) + self.custom_timesteps = True + else: + if num_inference_steps <= 0: + raise ValueError( + f"`num_inference_steps` must be a positive integer, but got {num_inference_steps}." + ) + if num_inference_steps > self.config.num_train_timesteps: + raise ValueError( + f"`num_inference_steps`: {num_inference_steps} cannot be larger than `self.config.train_timesteps`:" diff --git a/src/diffusers/schedulers/scheduling_ddpm.py b/src/diffusers/schedulers/scheduling_ddpm.py index 972c46c6e930..b2a335cd2e05 100644 --- a/src/diffusers/schedulers/scheduling_ddpm.py +++ b/src/diffusers/schedulers/scheduling_ddpm.py @@ -308,6 +308,10 @@ def set_timesteps( timesteps = np.array(timesteps, dtype=np.int64) self.custom_timesteps = True else: + if num_inference_steps <= 0: + raise ValueError( + f"`num_inference_steps` must be a positive integer, but got {num_inference_steps}." + ) if num_inference_steps > self.config.num_train_timesteps: raise ValueError( f"`num_inference_steps`: {num_inference_steps} cannot be larger than `self.config.train_timesteps`:"