tput instead of
a plain tuple.


jit (bool, defaults to False) —
Whether to run pmap versions of the generation and safety scoring functions. NOTE: 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



FlaxStableDiffusionPipelineOutput if return_dict is True, otherwise a
tuple. When returning a tuple, the first element is a list wit