 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.Optional[typing.Callable] = None

)


Parameters 

attention_op (Callable, optional) —
Override the default None operator for use as op argument to the
memory_efficient_attention()
function of xFormers.



Enable memory efficient attention as