e of slice_size.



Enable sliced attention computation.
When this option is enabled, the attention 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.

enable_xformers_memory_efficient_attention

<
source
>
(
attention_op: typing.Optional[typing.Callable] = None

)


Parameters 

attention_op (Callable, optional) —
Override the default N