update get_order_list if statement (#7309)

* update get_order_list if statement * revery

update get_order_list if statement (#7309)
* update get_order_list if statement * revery
0b45b588 · Kenneth Gerald Hamilton · GitHub · d3986f18 · 0b45b588
Unverified Commit 0b45b588 authored Mar 14, 2024 by Kenneth Gerald Hamilton Committed by GitHub Mar 13, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py +2 -0

No files found.
--- a/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py
+++ b/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py
@@ -223,6 +223,8 @@ class DPMSolverSinglestepScheduler(SchedulerMixin, ConfigMixin):
        """
        steps = num_inference_steps
        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 order == 3:
                if steps % 3 == 0: