Rate this Page
★ ★ ★ ★ ★

torch.nn.attention.restore_flash_attention_impl#

torch.nn.attention.restore_flash_attention_impl(_raise_warn=True)[source]#

Restore the default FA2 implementation

On this page
Show Source
PyTorch Libraries