Skip to content

Commit 0b45b58

Browse files
authored
update get_order_list if statement (#7309)
* update get_order_list if statement * revery
1 parent d3986f1 commit 0b45b58

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,8 @@ def get_order_list(self, num_inference_steps: int) -> List[int]:
223223
"""
224224
steps = num_inference_steps
225225
order = self.config.solver_order
226+
if order > 3:
227+
raise ValueError("Order > 3 is not supported by this scheduler")
226228
if self.config.lower_order_final:
227229
if order == 3:
228230
if steps % 3 == 0:

0 commit comments

Comments
 (0)