Skip to content

GitLab

Unverified Commit 0b45b588 authored by Kenneth Gerald Hamilton's avatar Kenneth Gerald Hamilton Committed by GitHub
Browse files

update get_order_list if statement (#7309) 

* update get_order_list if statement

* revery
parent d3986f18
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py
View file @ 0b45b588
...@@ -223,6 +223,8 @@ class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin): ...@@ -223,6 +223,8 @@ class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin):
""" """
steps = num_inference_steps steps = num_inference_steps
order = self.config.solver_order order = self.config.solver_order
if order > 3:
raise ValueError("Order > 3 is not supported by this scheduler")
if self.config.lower_order_final: if self.config.lower_order_final:
if order == 3: if order == 3:
if steps % 3 == 0: if steps % 3 == 0:
......
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