fix lint

yingtongxiong · Sep 4, 2023 · 720e4c8 · 720e4c8
1 parent 8a4e650
commit 720e4c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internlm/model/linear.py b/internlm/model/linear.py
@@ -9,7 +9,7 @@
 from flash_attn.utils.distributed import all_reduce, reduce_scatter
 from torch import nn
 
-from internlm.core.context import IS_TENSOR_PARALLEL, ParallelMode
+from internlm.core.context import ParallelMode
 from internlm.core.context import global_context as gpc
 from internlm.model.utils import fused_dense_func_torch