Skip to content

GitLab

"src/diffusers/utils/hub_utils.py" did not exist on "71289ba06ea897270ad6de0ea7ff641f4a7b246c"
Commit df64f624 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

finish pndm

parent f0a99e76
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/schedulers/scheduling_pndm.py
View file @ df64f624
...@@ -96,6 +96,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin): ...@@ -96,6 +96,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
return self.time_steps[num_inference_steps] return self.time_steps[num_inference_steps]
def step_warm_up(self, residual, image, t, num_inference_steps): def step_warm_up(self, residual, image, t, num_inference_steps):
# TODO(Patrick) - need to rethink whether the "warmup" way is the correct API design here
warmup_time_steps = self.get_warmup_time_steps(num_inference_steps) warmup_time_steps = self.get_warmup_time_steps(num_inference_steps)
t_prev = warmup_time_steps[t // 4 * 4] t_prev = warmup_time_steps[t // 4 * 4]
......
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