Skip to content

Commit bd11bf0

Browse files
change turing names7
1 parent 134f4fd commit bd11bf0

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

library/attention.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,10 +17,10 @@
1717

1818
if flash_attn is None:
1919
try:
20-
import flash_attn_turing as flash_attn
21-
from flash_attn.flash_attention_interface import _flash_attn_forward
22-
from flash_attn.flash_attention_interface import flash_attn_varlen_func
23-
from flash_attn.flash_attention_interface import flash_attn_func
20+
import flash_attn_turing
21+
from flash_attn_turing.flash_attention_interface import _flash_attn_forward
22+
from flash_attn_turing.flash_attention_interface import flash_attn_varlen_func
23+
from flash_attn_turing.flash_attention_interface import flash_attn_func
2424
except ImportError as e:
2525
flash_attn = None
2626
flash_attn_varlen_func = None

0 commit comments

Comments
 (0)