From f6f055a93df4fb1a59fa8f28e3b8f092fd7ba511 Mon Sep 17 00:00:00 2001
From: v0xie <28695009+v0xie@users.noreply.github.com>
Date: Fri, 19 Jul 2024 14:08:44 -0700
Subject: [PATCH] use configured alpha/beta values in Beta scheduling

---
 modules/sd_schedulers.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/modules/sd_schedulers.py b/modules/sd_schedulers.py
index 6f2dc6308..f4d16e309 100644
--- a/modules/sd_schedulers.py
+++ b/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]