Update diffusion_webui/diffusion_models/controlnet_pipeline.py

diffusion_webui/diffusion_models/controlnet_pipeline.py

@@ -71,8 +71,13 @@ class StableDiffusionControlNetGenerator(ControlnetPipeline):
             controlnet_model_path=controlnet_model_path,
             scheduler=scheduler,
         )
+        if preprocces_type== "ScribbleXDOG":
+            read_image = cv2.imread(image_path)
+            controlnet_image = Image.fromarray(read_image)
+
+        else:
+            read_image = Image.open(image_path)
 
-        read_image = cv2.imread(image_path)
         controlnet_image = self.controlnet_preprocces(
             read_image=read_image, preprocces_type=preprocces_type
         )[0]
@@ -83,7 +88,6 @@ class StableDiffusionControlNetGenerator(ControlnetPipeline):
         else:
             generator = torch.manual_seed(seed_generator)
         
-        controlnet_image = Image.fromarray(controlnet_image)
 
         output = pipe(
             prompt=prompt,