Ver Fonte

use configured alpha/beta values in Beta scheduling

v0xie há 1 ano atrás
pai
commit
f6f055a93d
1 ficheiros alterados com 2 adições e 2 exclusões
  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):
     # 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 = [stats.beta.ppf(x, alpha, beta) for x in timesteps]
     sigmas = [sigma_min + (x * (sigma_max-sigma_min)) for x in timesteps]