kadirnar commited on
Commit
860a671
1 Parent(s): 8ec9be7

Update diffusion_webui/diffusion_models/controlnet_pipeline.py

Browse files
diffusion_webui/diffusion_models/controlnet_pipeline.py CHANGED
@@ -105,7 +105,9 @@ class StableDiffusionControlNetGenerator(ControlnetPipeline):
105
  read_image = cv2.imread(image_path)
106
  controlnet_image = self.controlnet_preprocces(read_image=read_image, preprocces_type=preprocces_type)[0]
107
  controlnet_image = Image.fromarray(controlnet_image)
108
-
 
 
109
  else:
110
  read_image = Image.open(image_path)
111
  controlnet_image = self.controlnet_preprocces(read_image=read_image, preprocces_type=preprocces_type)
 
105
  read_image = cv2.imread(image_path)
106
  controlnet_image = self.controlnet_preprocces(read_image=read_image, preprocces_type=preprocces_type)[0]
107
  controlnet_image = Image.fromarray(controlnet_image)
108
+
109
+ elif preprocces_type== "None":
110
+ controlnet_image = self.controlnet_preprocces(read_image=image_path, preprocces_type=preprocces_type)
111
  else:
112
  read_image = Image.open(image_path)
113
  controlnet_image = self.controlnet_preprocces(read_image=read_image, preprocces_type=preprocces_type)