_ids = shard(prompt_ids)

>>> images = pipeline(prompt_ids, params, prng_seed, num_inference_steps, jit=True).images
>>> images = pipeline.numpy_to_pil(np.asarray(images.reshape((num_samples,) + images.shape[-3:])))  FlaxStableDiffusionPipelineOutput  class diffusers.pipelines.stable_diffusion.FlaxStableDiffusionPipelineOutput  < source > ( images: ndarray nsfw_content_detected: List  )   Parameters   images (np.ndarray) —
Denoised images of ar