Re-add explicit device move

This commit is contained in:
random_thoughtss 2022-10-29 14:13:02 -07:00
parent 6e2ce4e735
commit 39f55c3c35

View file

@ -170,6 +170,7 @@ class StableDiffusionProcessing():
# Create another latent image, this time with a masked version of the original input.
# Smoothly interpolate between the masked and unmasked latent conditioning image using a parameter.
conditioning_mask = conditioning_mask.to(source_image.device).to(source_image.dtype)
conditioning_image = torch.lerp(
source_image,
source_image * (1.0 - conditioning_mask),