Fixed error that occurs when using vanilla samplers (somehow).

This commit is contained in:
CodeHatchling 2023-12-07 14:54:26 -07:00
parent 56604f08a1
commit 0ef4a4cb23

View File

@ -937,8 +937,8 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
p.scripts.postprocess_image(p, pp) p.scripts.postprocess_image(p, pp)
image = pp.image image = pp.image
mask_for_overlay = p.mask_for_overlay mask_for_overlay = getattr(p, "mask_for_overlay", None)
overlay_image = p.overlay_images[i] if p.overlay_images is not None and i < len(p.overlay_images) else None overlay_image = p.overlay_images[i] if getattr(p, "overlay_images", None) is not None and i < len(p.overlay_images) else None
if p.scripts is not None: if p.scripts is not None:
ppmo = scripts.PostProcessMaskOverlayArgs(i, mask_for_overlay, overlay_image) ppmo = scripts.PostProcessMaskOverlayArgs(i, mask_for_overlay, overlay_image)