aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2025-12-21 15:24:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Making spacing function slimmer

Browse Source
This commit is contained in:
Tim
2023-06-30 09:09:32 -07:00
parent e9869d7822
commit 2796c81a5f
1 changed files with 4 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
sgm/modules/diffusionmodules/discretizer.py
View File

@@ -6,24 +6,10 @@ from ...util import append_zero
from ...modules.diffusionmodules.util import make_beta_schedule from ...modules.diffusionmodules.util import make_beta_schedule
def generate_roughly_equally_spaced_steps(n, m): def generate_roughly_equally_spaced_steps(
# 0, ..., m - 1 num_substeps: int, max_step: int
m -= 1 ) -> np.ndarray:
return np.linspace(max_step - 1, 0, num_substeps, endpoint=False).astype(int)[::-1]
# We are getting rid of leading 0 later, so increase steps
n += 1
# Calculate the step size
step = m / (n - 1)
# Generate the list
steps_reversed = [int(m - i * step) for i in range(n)]
steps = steps_reversed[::-1]
# Get rid of leading 0
steps = steps[1:]
return np.array(steps)
class Discretization: class Discretization: