[Bug] Using all-in-one prompt prevents normal image grids from being generated

[wiseoldowl-66]

Issue Feedback

• [X] I confirm that I have searched for a solution to this issue in the FAQ and couldn't find a solution.
• [X] I confirm that I have searched for this issue in the Issues list (including closed ones) and couldn't find a solution.
• [X] I confirm that I have read the Wiki and couldn't find a solution.

Describe the Issue

Under normal circumstances when more than one image is batch generated, either through batch count or batch size >1, A1111 produces a single image grid of all generated images.

When using the all-in-one prompt, this grid is not generated at all with a batch count of >1 and batch size of 1.

If both count and size are >1, multiple grids are produced with the number of images in each corresponding to batch size.

The grid is generated normally when the extension is inactive.

Steps to Reproduce

Produce any images using the extension in a batch count of more than one.

Screenshot or log

No warnings in console.

Extension unloaded:

Extension loaded with batch size 1:

Extension loaded with batch size >1:

OS

Windows

Browser

Edge

Stable Diffusion WebUI version

1.8.0

Extension version

2cd8bf2c

Python version

3.10.6

Gradio version

3.41.2