22

app.py

@@ -127,12 +127,13 @@ def inference(
     input_image, width, height = image_utils.rescale_image(input_image, 1, 8, max_size=1280)
     sample_size = [height, width]
     
-    print("DEBUG: control_image_torch")
+    print("DEBUG: inpaint_image")
     if input_image is not None:
         inpaint_image = get_image_latent(input_image, sample_size=sample_size)[:, :, 0]
     else:
         inpaint_image = torch.zeros([1, 3, sample_size[0], sample_size[1]])
 
+    print("DEBUG: mask_image")
     if mask_image is not None:
         mask_image = image_utils.rescale_image(mask_image, 1, 8, max_size=1280)
         mask_image = get_image_latent(mask_image, sample_size=sample_size)[:, :1, 0]