Spaces:
Building
on
A10G
Building
on
A10G
Update diffusion_webui/diffusion_models/controlnet_pipeline.py
Browse files
diffusion_webui/diffusion_models/controlnet_pipeline.py
CHANGED
@@ -75,7 +75,7 @@ class StableDiffusionControlNetGenerator(ControlnetPipeline):
|
|
75 |
read_image = cv2.imread(image_path)
|
76 |
controlnet_image = self.controlnet_preprocces(
|
77 |
read_image=read_image, preprocces_type=preprocces_type
|
78 |
-
)
|
79 |
|
80 |
if seed_generator == 0:
|
81 |
random_seed = torch.randint(0, 1000000, (1,))
|
|
|
75 |
read_image = cv2.imread(image_path)
|
76 |
controlnet_image = self.controlnet_preprocces(
|
77 |
read_image=read_image, preprocces_type=preprocces_type
|
78 |
+
)[0]
|
79 |
|
80 |
if seed_generator == 0:
|
81 |
random_seed = torch.randint(0, 1000000, (1,))
|