always add alpha/beta to extra_generation_params when schedule is Beta

2024-07-19 14:54:24 -07:00 · 2024-07-19 14:54:24 -07:00 · 9de7084884
parent 94275b115c
commit 9de7084884
1 changed files with 4 additions and 0 deletions
--- a/modules/sd_samplers_kdiffusion.py
+++ b/modules/sd_samplers_kdiffusion.py
@ -119,6 +119,10 @@ class KDiffusionSampler(sd_samplers_common.Sampler):

            if scheduler.need_inner_model:
                sigmas_kwargs['inner_model'] = self.model_wrap
+            
+            if scheduler.label == 'Beta':
+                p.extra_generation_params["Beta schedule alpha"] = opts.beta_dist_alpha
+                p.extra_generation_params["Beta schedule beta"] = opts.beta_dist_beta

            sigmas = scheduler.function(n=steps, **sigmas_kwargs, device=devices.cpu)