Skip to content

S
Stable Diffusion Webui
Commit 7e1bd3e3 authored by v0xie's avatar v0xie
Browse files
Options

refactor: syntax and add 0.0 on new line

parent a5f66b50
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
modules/sd_schedulers.py
View file @ 7e1bd3e3
...@@ -122,9 +122,9 @@ def beta_scheduler(n, sigma_min, sigma_max, inner_model, device): ...@@ -122,9 +122,9 @@ def beta_scheduler(n, sigma_min, sigma_max, inner_model, device):
beta = 0.6 beta = 0.6
timesteps = 1 - np.linspace(0, 1, n) timesteps = 1 - np.linspace(0, 1, n)
timesteps = [stats.beta.ppf(x, alpha, beta) for x in timesteps] timesteps = [stats.beta.ppf(x, alpha, beta) for x in timesteps]
sigmas = [sigma_min + ((x)*(sigma_max-sigma_min)) for x in timesteps] + [0.0] sigmas = [sigma_min + (x * (sigma_max-sigma_min)) for x in timesteps]
sigmas = torch.FloatTensor(sigmas).to(device) sigmas += [0.0]
return sigmas return torch.FloatTensor(sigmas).to(device)
schedulers = [ schedulers = [
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment