n. Can be used to tweak the same generation with different prompts. If not provided, a latents
array is generated by sampling using the supplied random generator.   jit (bool, defaults to False) —
Whether to run pmap versions of the generation and safety scoring functions.

This argument exists because __call__ is not yet end-to-end pmap-able. It will be removed in a
future release.
   return_dict (bool, optional, defaults to True) —
Whether or