Skip to content

Commit 2e8e6cc

Browse files
committed
Fix wrong output on 3n-1 steps count
1 parent 926daa3 commit 2e8e6cc

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,10 @@ def get_order_list(self, num_inference_steps: int) -> List[int]:
248248
orders = [1, 2] * (steps // 2)
249249
elif order == 1:
250250
orders = [1] * steps
251+
252+
if self.config.final_sigmas_type == "zero":
253+
orders[-1] = 1
254+
251255
return orders
252256

253257
@property

0 commit comments

Comments
 (0)