Commit 39f55c3c authored by random_thoughtss's avatar random_thoughtss

Re-add explicit device move

parent 6e2ce4e7
...@@ -170,6 +170,7 @@ class StableDiffusionProcessing(): ...@@ -170,6 +170,7 @@ class StableDiffusionProcessing():
# Create another latent image, this time with a masked version of the original input. # 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. # 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( conditioning_image = torch.lerp(
source_image, source_image,
source_image * (1.0 - conditioning_mask), source_image * (1.0 - conditioning_mask),
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment