diff --git a/src/diffusers/pipeline_flax_utils.py b/src/diffusers/pipeline_flax_utils.py index 0ef2f166edfc..848174230287 100644 --- a/src/diffusers/pipeline_flax_utils.py +++ b/src/diffusers/pipeline_flax_utils.py @@ -498,7 +498,9 @@ def components(self) -> Dict[str, Any]: ... FlaxStableDiffusionImg2ImgPipeline, ... ) - >>> text2img = FlaxStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", revision="bf16", dtype=jnp.bfloat16) + >>> text2img = FlaxStableDiffusionPipeline.from_pretrained( + ... "runwayml/stable-diffusion-v1-5", revision="bf16", dtype=jnp.bfloat16 + ... ) >>> img2img = FlaxStableDiffusionImg2ImgPipeline(**text2img.components) ```