Note there is some small visual difference between num_images_per_prompt=1 and num_images_per_prompt=2:

num_images_per_prompt=1

num_images_per_prompt=2

This also occurs without skip_guidance_layers:

num_images_per_prompt=1

num_images_per_prompt=2

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

can we add more tests here

feel free to merge once tested is add (or you can add test in a separate PR!)

Yes it's easily missed, important to check batch support for new models etc. going forward.

Looks like tests for num_images_per_prompt are missing from some other pipelines like Flux as well so I'll add them all in another PR.

Looks like tests for num_images_per_prompt are missing from some other pipelines like Flux as well so I'll add them all in another PR.

@hlky

I think it would make sense to introduce/revisit the pipeline tester mixin to see how we can make it more general so that this is tested by default without us having to write it for every pipeline/model test suite. WDYT?

Feel free to ping me and @DN6 if you need further ideas here.