Spaces:
Paused
Paused
Update diffusion_webui/diffusion_models/controlnet/controlnet_inpaint/pipeline_stable_diffusion_controlnet_inpaint.py
Browse files
diffusion_webui/diffusion_models/controlnet/controlnet_inpaint/pipeline_stable_diffusion_controlnet_inpaint.py
CHANGED
@@ -452,15 +452,15 @@ class StableDiffusionControlNetInpaintPipeline(
|
|
452 |
|
453 |
# 4. Prepare image
|
454 |
control_image = self.prepare_image(
|
455 |
-
|
456 |
-
|
457 |
-
|
458 |
-
|
459 |
-
|
460 |
-
|
461 |
-
|
462 |
-
|
463 |
-
|
464 |
if do_classifier_free_guidance:
|
465 |
control_image = torch.cat([control_image] * 2)
|
466 |
|
|
|
452 |
|
453 |
# 4. Prepare image
|
454 |
control_image = self.prepare_image(
|
455 |
+
control_image,
|
456 |
+
width,
|
457 |
+
height,
|
458 |
+
batch_size * num_images_per_prompt,
|
459 |
+
num_images_per_prompt,
|
460 |
+
device,
|
461 |
+
do_classifier_free_guidance,
|
462 |
+
self.controlnet.dtype,
|
463 |
+
)
|
464 |
if do_classifier_free_guidance:
|
465 |
control_image = torch.cat([control_image] * 2)
|
466 |
|