Bläddra i källkod

use configured alpha/beta values in Beta scheduling

v0xie 1 år sedan
förälder
incheckning
f6f055a93d
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      modules/sd_schedulers.py

+ 2 - 2
modules/sd_schedulers.py

@@ -118,8 +118,8 @@ def ddim_scheduler(n, sigma_min, sigma_max, inner_model, device):
 
 
 def beta_scheduler(n, sigma_min, sigma_max, inner_model, device):
 def beta_scheduler(n, sigma_min, sigma_max, inner_model, device):
     # From "Beta Sampling is All You Need" [arXiv:2407.12173] (Lee et. al, 2024) """
     # From "Beta Sampling is All You Need" [arXiv:2407.12173] (Lee et. al, 2024) """
-    alpha = 0.6
-    beta = 0.6
+    alpha = shared.opts.beta_dist_alpha
+    beta = shared.opts.beta_dist_beta
     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]
     sigmas = [sigma_min + (x * (sigma_max-sigma_min)) for x in timesteps]