Re-add explicit device move

2022-10-29 14:13:02 -07:00 · 2022-10-29 14:13:02 -07:00 · 39f55c3c35
parent 6e2ce4e735
commit 39f55c3c35
1 changed files with 1 additions and 0 deletions
--- a/modules/processing.py
+++ b/modules/processing.py
@ -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),