..

mosaicml · Aug 15, 2024 · de76124 · de76124
1 parent 0f44dae
commit de76124
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/models/layers/test_attention.py b/tests/models/layers/test_attention.py
@@ -8,7 +8,6 @@
 
 from llmfoundry.models.layers.attention import (
     attention_implementations,
-    flash_attn_fn,
     scaled_multihead_dot_product_attention,
 )
 from llmfoundry.models.layers.layer_builders import build_attention_layer