ng” or “checkpoint
activations”.

disable_xformers_memory_efficient_attention

<
source
>
(
)



Disable memory efficient attention as implemented in xformers.

enable_gradient_checkpointing

<
source
>
(
)



Activates gradient checkpointing for the current model.
Note that in other frameworks this feature can be referred to as “activation checkpointing” or “checkpoint
activations”.

enable_xformers_memory_efficient_attention

<
source
>
(
att