Update modules/model.py

modules/model.py

@@ -367,8 +367,12 @@ class StableDiffusionPipeline(DiffusionPipeline):
             ).input_ids
 
             dotmap = v["map"] < 255
+            out = dotmap.astype(float)
+            if v["mask_outsides"]:
+                out[out==0] = -1
+                
             arr = torch.from_numpy(
-                dotmap.astype(float) * float(v["weight"]) * g_strength
+                out * float(v["weight"]) * g_strength
             )
             img_state.append((v_input, arr))
 
@@ -599,6 +603,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
         # 2. Define call parameters
         batch_size = 1 if isinstance(prompt, str) else len(prompt)
         device = self._execution_device
+        latents = latents.to(device, dtype=self.unet.dtype)
         # here `guidance_scale` is defined analog to the guidance weight `w` of equation (2)
         # of the Imagen paper: https://arxiv.org/pdf/2205.11487.pdf . `guidance_scale = 1`
         # corresponds to doing no classifier free guidance.