clean unused import

OpenBMB · Jun 11, 2024 · 1614ad1 · 1614ad1
1 parent 5c9eacd
commit 1614ad1
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 62 deletions.
diff --git a/tests/test_burst.py b/tests/test_burst.py
@@ -1,6 +1,5 @@
 import torch
 import bmtrain as bmt
-from flash_attn.flash_attn_interface import flash_attn_func as flash_cuda
 import numpy as np
 
 OpBurstAttn = bmt.nn.OpBurstAttn

diff --git a/tests/test_lse.py b/tests/test_lse.py