has its forwardmethod called. Note that offloading happens on a submodule basis. Memory savings are higher than withenable_model_cpu_offload`, but performance is lower.

class diffusers.FlaxStableDiffusionImg2ImgPipeline

<
source
>
(
vae: FlaxAutoencoderKL
text_encoder: FlaxCLIPTextModel
tokenizer: CLIPTokenizer
unet: FlaxUNet2DConditionModel
scheduler: typing.Union[diffusers.schedulers.scheduling_ddim_flax.FlaxDDIMScheduler, diffusers.schedul