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