 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 not to return a FlaxStableDiffusionPipelineOutput instead of
a plain tuple.   Returns 
FlaxStableDiffusionPipelineOutput or tuple
  
If return_dict is True, FlaxStableDiffusionPipelin