Update __init__.py

sdtblck · web-flow · commit e4689e2ebc74 · 2021-05-02T12:44:55.000+02:00
diff --git a/deepspeed/ops/sparse_attention/__init__.py b/deepspeed/ops/sparse_attention/__init__.py
@@ -1,4 +1,4 @@
-from .sparsity_config import SparsityConfig, DenseSparsityConfig, FixedSparsityConfig, VariableSparsityConfig, BigBirdSparsityConfig, BSLongformerSparsityConfig
+from .sparsity_config import SparsityConfig, DenseSparsityConfig, FixedSparsityConfig, VariableSparsityConfig, BigBirdSparsityConfig, BSLongformerSparsityConfig, LocalSlidingWindowSparsityConfig
 from .softmax import Softmax
 from .matmul import MatMul
 from .sparse_self_attention import SparseSelfAttention

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from .sparsity_config import SparsityConfig, DenseSparsityConfig, FixedSparsityConfig, VariableSparsityConfig, BigBirdSparsityConfig, BSLongformerSparsityConfig`
	`1`	`+from .sparsity_config import SparsityConfig, DenseSparsityConfig, FixedSparsityConfig, VariableSparsityConfig, BigBirdSparsityConfig, BSLongformerSparsityConfig, LocalSlidingWindowSparsityConfig`
`2`	`2`	`from .softmax import Softmax`
`3`	`3`	`from .matmul import MatMul`
`4`	`4`	`from .sparse_self_attention import SparseSelfAttention`