—
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.
   Returns 
FlaxStableDiffusionPipelineOutput or tuple
  
If return_dict is True, FlaxStableDiffusionPipelineOutput is
returned, otherwise a tuple is returned where the first element is a list with the generated images
and the second element is a list of bo