This commit is contained in:
v0xie 2024-07-19 15:33:07 -07:00
parent 9de7084884
commit 8749540602
1 changed files with 1 additions and 1 deletions

View File

@ -119,7 +119,7 @@ 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': if scheduler.label == 'Beta':
p.extra_generation_params["Beta schedule alpha"] = opts.beta_dist_alpha p.extra_generation_params["Beta schedule alpha"] = opts.beta_dist_alpha
p.extra_generation_params["Beta schedule beta"] = opts.beta_dist_beta p.extra_generation_params["Beta schedule beta"] = opts.beta_dist_beta