vice('meta') and loaded to GPU only when their specific submodule 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.FlaxStableDiffusionPipeline

<
source
>
(
vae: FlaxAutoencoderKL
text_encoder: FlaxCLIPTextModel
tokenizer: CLIPTokenizer
unet: FlaxUNet2DConditionModel
scheduler: typing.Union[diffusers.scheduler