yindo/flash-attention
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention.git synced 2026-07-01 21:04:02 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix import error

Browse Source 
Fix #1618
This commit is contained in:
Tri Dao
2025-04-25 14:48:27 -04:00
parent 93690e2ab7
commit 41a21d6204
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
flash_attn/modules/mha.py
+1 -1
View File
@@ -25,7 +25,7 @@ except ImportError:
try:
from flash_attn.ops.fused_dense import ColumnParallelLinear, RowParallelLinear
except ImportError:
ColumnParallelLinear, RowParallelLinear = None, None, None
ColumnParallelLinear, RowParallelLinear = None, None
try:
from flash_attn.layers.rotary import RotaryEmbedding