Commit 9de70848 authored by v0xie's avatar v0xie

always add alpha/beta to extra_generation_params when schedule is Beta

parent 94275b11
...@@ -119,6 +119,10 @@ class KDiffusionSampler(sd_samplers_common.Sampler): ...@@ -119,6 +119,10 @@ class KDiffusionSampler(sd_samplers_common.Sampler):
if scheduler.need_inner_model: if scheduler.need_inner_model:
sigmas_kwargs['inner_model'] = self.model_wrap 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) sigmas = scheduler.function(n=steps, **sigmas_kwargs, device=devices.cpu)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment