diff --git a/ldm/models/diffusion/dpm_solver/dpm_solver.py b/ldm/models/diffusion/dpm_solver/dpm_solver.py index f5f59b9..79390c0 100644 --- a/ldm/models/diffusion/dpm_solver/dpm_solver.py +++ b/ldm/models/diffusion/dpm_solver/dpm_solver.py @@ -522,15 +522,15 @@ class DPM_Solver: if order == 3: K = steps // 3 + 1 if steps % 3 == 0: - orders = [3,] * ( + orders = [3, ] * ( K - 2 ) + [2, 1] elif steps % 3 == 1: - orders = [3,] * ( + orders = [3, ] * ( K - 1 ) + [1] else: - orders = [3,] * ( + orders = [3, ] * ( K - 1 ) + [2] elif order == 2: @@ -541,7 +541,7 @@ class DPM_Solver: ] * K else: K = steps // 2 + 1 - orders = [2,] * ( + orders = [2, ] * ( K - 1 ) + [1] elif order == 1: