kadirnar commited on
Commit
f17e558
1 Parent(s): 828cdd9

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
- control_image,
456
- width,
457
- height,
458
- batch_size * num_images_per_prompt,
459
- num_images_per_prompt,
460
- device,
461
- self.controlnet.dtype,
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