Spaces:
Runtime error
Runtime error
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)
|