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

Fix int64 with UniPC && Add OneAPI version logging

Browse Source
This commit is contained in:
Disty0
2023-05-28 20:00:03 +03:00
parent 7254925dca
commit 09141ee1a8
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_samplers_compvis.py
+4 -1
View File
@@ -97,7 +97,10 @@ class VanillaStableDiffusionSampler:
unconditional_conditioning = unconditional_conditioning[:, :cond.shape[1]]
if self.mask is not None:
img_orig = self.sampler.model.q_sample(self.init_latent, ts)
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)
x = img_orig * self.mask + self.nmask * x
# Wrap the image conditioning back up since the DDIM code can accept the dict directly.