Img2Img tag doesn't work with batches -- Fixed -- Code attached for new shortcode that allows img2img with batches

[pmajor74]

I've been reviewing the img2img tag and trying to get it to work with batches. Behind the scenes I can see that your img2img tag eventually calls the "modules.img2img.img2img" and always has "mode = 0" from what I can see. Though batch processing happens when mode=5 (at least inside the automatic1111 code for the img2img module).

I can't really see any shortcodes / tags developed to help with this issue currently. I've made a simple example template:

Photo of a a blonde haired slender woman.

[log INFO]Example: About to go into the after block[/log]

[after]
[log INFO]## BEFORE calling img2img[/log]
[img2img]*
[log INFO]## AFTER calling img2img[/log]
[/after]

When I call this in the prompt with [call example], with a batch size of 2, I get many more than 2 images output.

What I'm thinking of doing is possibly trying to change the output folder of the txt2img generation temporarily, then in the img2img tag if I detect a batch is occuring, then pass mode=5 to the a1111 txt2img module along with input and output paths and hope that the internal img2img does its thing as it has a process_batch that gets called with mode=5.

I've done this code update, but it just gives me even more images. I suspect because the [after] tag is batch aware and runs the internal code the number of batches specified, but the internal img2img isn't quite getting it.

Any advice on this so that I can update this to support batches in the img2img?

PS. I purchased your Beautiful Soul collection. Very nice. I have one comment, the download link gives you 3 ZIP files, each with apparently a different version of the Beautiful Soul collection. Is there any reason to install the 3 versions or is the latest one the only one needed?

[pmajor74]

After a lot of digging around, I've put together something that appears to work well enough so that you can have img2img output when you want to run batches (currently if you use the img2img tag you will be unable to do batches without some very strange output).

You can review the code solution and example of how to use the short code on the following pull request:
A fork / pull request has been made for this here: #232