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 #889 - fuzzy k* img2img at low strength

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-06 14:16:56 -04:00
parent 74419f41a3
commit 2c27e759cd
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -127,13 +127,12 @@ class KSampler(Sampler):
# sigmas are now set up in make_schedule - we take the last steps items # sigmas are now set up in make_schedule - we take the last steps items
sigmas = self.sigmas[-S-1:] sigmas = self.sigmas[-S-1:]
x = torch.randn([batch_size, *shape], device=self.device) * sigmas[0]
# for GPU draw
if x_T is not None: if x_T is not None:
x = x_T * sigmas[0] x = x_T + x
else:
x = (
torch.randn([batch_size, *shape], device=self.device)
* sigmas[0]
) # for GPU draw
model_wrap_cfg = CFGDenoiser(self.model, threshold=threshold, warmup=max(0.8*S,S-10)) model_wrap_cfg = CFGDenoiser(self.model, threshold=threshold, warmup=max(0.8*S,S-10))
extra_args = { extra_args = {
'cond': conditioning, 'cond': conditioning,