Skip to content

Commit 39f55c3

Browse files
Re-add explicit device move
1 parent 6e2ce4e commit 39f55c3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

modules/processing.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,7 @@ def img2img_image_conditioning(self, source_image, latent_image, image_mask = No
170170

171171
# Create another latent image, this time with a masked version of the original input.
172172
# Smoothly interpolate between the masked and unmasked latent conditioning image using a parameter.
173+
conditioning_mask = conditioning_mask.to(source_image.device).to(source_image.dtype)
173174
conditioning_image = torch.lerp(
174175
source_image,
175176
source_image * (1.0 - conditioning_mask),

0 commit comments

Comments
 (0)