    return {"prompt": prompts, "generator": generator, "num_inference_steps": num_inference_steps} Start with batch_size=4 and see how much memory you’ve consumed:   Copied from diffusers.utils import make_image_grid