22

app.py

@@ -145,13 +145,13 @@ def inference(
     control_image = control_image.resize((width, height))
     control_image_torch = get_image_latent(control_image, sample_size=sample_size)[:, :, 0]
 
-    # mask_image = None
+    mask_image = None
     # inpaint_image = None
     
-    # if mask_image is not None:
-    #     mask_image = get_image_latent(mask_image, sample_size=sample_size)[:, :1, 0]
-    # else:
-    #     mask_image = torch.ones([1, 1, sample_size[0], sample_size[1]]) * 255
+    if mask_image is not None:
+        mask_image = get_image_latent(mask_image, sample_size=sample_size)[:, :1, 0]
+    else:
+        mask_image = torch.ones([1, 1, sample_size[0], sample_size[1]]) * 255
     
     # if inpaint_image is not None:
     #     inpaint_image = get_image_latent(inpaint_image, sample_size=sample_size)[:, :, 0]
@@ -170,7 +170,7 @@ def inference(
         generator=generator,
         guidance_scale=guidance_scale,
         image               = None,
-        mask_image          = None,
+        mask_image          = mask_image,
         control_image=control_image_torch,
         num_inference_steps=num_inference_steps,
         control_context_scale=control_context_scale,