aussian distribution to be used as inputs for image
generation. Can be used to tweak the same generation with different prompts. The array is generated by
sampling using the supplied random generator.   return_dict (bool, optional, defaults to True) —
Whether or not to return a FlaxStableDiffusionPipelineOutput instead of
a plain tuple.   jit (bool, defaults to False) —
Whether to run pmap versions of the generation and safety scoring functions