run make_schedule() if it hasn't already been called

- fixes #984
This commit is contained in:
Lincoln Stein
2022-10-09 15:29:04 -04:00
parent 6e2fd2affe
commit c5344acb25
2 changed files with 17 additions and 0 deletions

View File

@ -171,6 +171,14 @@ class KSampler(Sampler):
if img_callback is not None:
img_callback(k_callback_values['x'],k_callback_values['i'])
# if make_schedule() hasn't been called, we do it now
if self.sigmas is None:
self.make_schedule(
ddim_num_steps=S,
ddim_eta = eta,
verbose = False,
)
# sigmas are set up in make_schedule - we take the last steps items
total_steps = len(self.sigmas)
sigmas = self.sigmas[-S-1:]