Fix img2img and hires for IPEX

This commit is contained in:
Disty0
2023-07-14 02:21:52 +03:00
parent d4c9d752f4
commit 558b71f088
3 changed files with 5 additions and 4 deletions
+1 -1
View File
@@ -216,7 +216,7 @@ class StableDiffusionProcessing:
conditioning_mask = torch.nn.functional.interpolate(conditioning_mask, size=latent_image.shape[-2:])
conditioning_mask = conditioning_mask.expand(conditioning_image.shape[0], -1, -1, -1)
image_conditioning = torch.cat([conditioning_mask, conditioning_image], dim=1)
image_conditioning = image_conditioning.to(device = shared.device, dtype = source_image.dtype)
image_conditioning = image_conditioning.to(device=shared.device, dtype=source_image.dtype)
return image_conditioning
def img2img_image_conditioning(self, source_image, latent_image, image_mask=None):