hw6

infer.py

@@ -137,6 +137,9 @@ def prepare_controlnet_image(controlnet_image, mode):
         image = np.repeat(image[:, :, None], 3, axis=2)
         image = Image.fromarray(image)
 
+    elif mode == "Pixel to Pixel":
+        image = controlnet_image.convert('RGB')
+
     elif mode == "HED edge detection (soft edge)":
         processor = HEDdetector.from_pretrained('lllyasviel/Annotators')
         image = processor(controlnet_image)