Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix unipc inpainting

Browse Source
This commit is contained in:
cool-bigdogs-tshirt
2023-06-01 17:48:28 -05:00
parent 562947c944
commit 1268d2762f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_compvis.py
+4 -4
View File
@@ -104,10 +104,10 @@ class VanillaStableDiffusionSampler:
unconditional_conditioning = unconditional_conditioning[:, :cond.shape[1]]
if self.mask is not None:
if shared.cmd_opts.use_ipex:
img_orig = self.sampler.model.q_sample(self.init_latent, ts.type(torch.int64))
else:
img_orig = self.sampler.model.q_sample(self.init_latent, ts)
encode_fn = self.sampler.model.q_sample
if self.is_unipc:
encode_fn = self.sampler.stochastic_encode
img_orig = encode_fn(self.init_latent, ts)
x = img_orig * self.mask + self.nmask * x
# Wrap the image conditioning back up since the DDIM code can accept the dict directly.