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 |
|