Return X images per text prompt instead of 1

[Radu Sibechi]

Hi,

First of all thanks for the implementation! Is it possible to return X images instead of just 1?

[Brett Kuprel]

Good idea

[Brett Kuprel]

I just updated the model to generate a grid of images. There's a demo in the colab.