update get_order_list if statement (#7309)

* update get_order_list if statement * revery
huggingface · Mar 14, 2024 · 0b45b58 · 0b45b58
1 parent d3986f1
commit 0b45b58
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py b/src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py
@@ -223,6 +223,8 @@ def get_order_list(self, num_inference_steps: int) -> List[int]:
         """
         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: