diff options
| author | random_thoughtss <random_thoughtss@proton.me> | 2022-10-29 14:13:02 -0700 |
|---|---|---|
| committer | random_thoughtss <random_thoughtss@proton.me> | 2022-10-29 14:13:02 -0700 |
| commit | 39f55c3c35873bc7dd9792cb2155746a1c3d4292 (patch) | |
| tree | 27916361c27af4772b81a5d9178786d7d3e0e3b3 /modules | |
| parent | 6e2ce4e735db64afcd0fe637327ca4ec78335706 (diff) | |
Re-add explicit device move
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/processing.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/processing.py b/modules/processing.py index ee0e9e34..d07e3db9 100644 --- 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),
|