Skip to content

GitLab

Commit cb90fd69 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

upload code

parent f794432e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/schedulers/scheduling_ddpm.py
View file @ cb90fd69
...@@ -94,6 +94,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): ...@@ -94,6 +94,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
self.set_format(tensor_format=tensor_format) self.set_format(tensor_format=tensor_format)
def set_timesteps(self, num_inference_steps): 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.num_inference_steps = num_inference_steps
self.timesteps = np.arange( self.timesteps = np.arange(
0, self.config.num_train_timesteps, self.config.num_train_timesteps // self.num_inference_steps 0, self.config.num_train_timesteps, self.config.num_train_timesteps // self.num_inference_steps
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment