Spaces:
Paused
Paused
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)
|