diff --git a/modules/sd_samplers_kdiffusion.py b/modules/sd_samplers_kdiffusion.py
index 95a354dac..4e5310bc6 100644
--- 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)