Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix crash on len(Nonetype) in k_sampler

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-07 08:05:13 -04:00
parent c6a7be63b8
commit 9f34ddfcea
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ldm/models/diffusion/ksampler.py
View File

@ -98,6 +98,7 @@ class KSampler(Sampler):
rho=7.,
device=self.device,
)
self.sigmas = self.karras_sigmas
# ALERT: We are completely overriding the sample() method in the base class, which
# means that inpainting will not work. To get this to work we need to be able to
@ -171,8 +172,8 @@ class KSampler(Sampler):
img_callback(k_callback_values['x'],k_callback_values['i'])
# sigmas are set up in make_schedule - we take the last steps items
total_steps = len(self.karras_sigmas)
sigmas = self.karras_sigmas[-S-1:]
total_steps = len(self.sigmas)
sigmas = self.sigmas[-S-1:]
if x_T is not None:
x = x_T + torch.randn([batch_size, *shape], device=self.device) * sigmas[0]