ntion module will split the input tensor in slices, to compute attention
in several steps. This is useful to save some memory in exchange for a small speed decrease.

disable_attention_slicing

<
source
>
(
)



Disable sliced attention computation. If enable_attention_slicing was previously invoked, this method will go
back to computing attention in one step.

enable_xformers_memory_efficient_attention

<
source
>
(
attention_op: typing.Option