JaehaerysNL/StableDiffusion
1
0
Fork 0
mirror of https://github.com/Stability-AI/stablediffusion.git synced 2024-12-23 08:04:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

Reformatting

Browse source
This commit is contained in:
Andres Caicedo 2023-05-08 14:33:14 +02:00
parent a523e6b9c1
commit a825f77092
No known key found for this signature in database
GPG key ID: 6E797C4F5A327624
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
ldm/models/diffusion/dpm_solver/dpm_solver.py
View file

@ -522,15 +522,15 @@ class DPM_Solver:
if order == 3: if order == 3:
K = steps // 3 + 1 K = steps // 3 + 1
if steps % 3 == 0: if steps % 3 == 0:
orders = [3,] * ( orders = [3, ] * (
K - 2 K - 2
) + [2, 1] ) + [2, 1]
elif steps % 3 == 1: elif steps % 3 == 1:
orders = [3,] * ( orders = [3, ] * (
K - 1 K - 1
) + [1] ) + [1]
else: else:
orders = [3,] * ( orders = [3, ] * (
K - 1 K - 1
) + [2] ) + [2]
elif order == 2: elif order == 2:
@ -541,7 +541,7 @@ class DPM_Solver:
] * K ] * K
else: else:
K = steps // 2 + 1 K = steps // 2 + 1
orders = [2,] * ( orders = [2, ] * (
K - 1 K - 1
) + [1] ) + [1]
elif order == 1: elif order == 1: