k-samplers fix 🔥

now THIS should fix it
This commit is contained in:
hlky 2022-08-25 04:19:55 +01:00
parent 07e4d738d5
commit 5240a47ee6
No known key found for this signature in database
GPG Key ID: 55A99F1E80D907D5

View File

@ -156,7 +156,7 @@ class KDiffusionSampler:
x = x_T * sigmas[0]
model_wrap_cfg = CFGDenoiser(self.model_wrap)
samples_ddim = K.sampling.sample_lms(model_wrap_cfg, x, sigmas, extra_args={'cond': conditioning, 'uncond': unconditional_conditioning, 'cond_scale': unconditional_guidance_scale}, disable=False)
samples_ddim = K.sampling.__dict__[f'sample_{self.schedule}'](model_wrap_cfg, x, sigmas, extra_args={'cond': conditioning, 'uncond': unconditional_conditioning, 'cond_scale': unconditional_guidance_scale}, disable=False)
return samples_ddim, None