[MHA] Run black on mha.py

bec5b3d3 · Tri Dao · cb0daccc · bec5b3d3
Commit bec5b3d3 authored Aug 16, 2023 by Tri Dao
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 374 additions and 175 deletions

flash_attn/modules/mha.py flash_attn/modules/mha.py +374 -175

No files found.
--- a/flash_attn/modules/mha.py
+++ b/flash_attn/modules/mha.py