oat16
... )

>>> pipe = pipe.to("cuda")

>>> prompt = "a photo of the dolomites"
>>> image = pipe(prompt).images[0]

disable_vae_slicing

<
source
>
(
)



Disable sliced VAE decoding. If enable_vae_slicing was previously invoked, this method will go back to
computing decoding in one step.

enable_sequential_cpu_offload

<
source
>
(
gpu_id = 0

)



Offloads all models to CPU using accelerate, significantly reducing memory usage. When called, 