..

mosaicml · Nov 18, 2023 · 88c6808 · 88c6808
1 parent f02034b
commit 88c6808
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -519,7 +519,7 @@ def test_mpt_creation(norm_type: str, no_bias: bool, tie_word_embeddings: bool):
         assert block.resid_ffn_dropout.p == 0.2
 
 
-@pytest.mark.parametrize('attn_impl', [
+@pytest.mark.parametrize('attention_impl', [
     'torch',
     pytest.param('flash', marks=pytest.mark.gpu),
     pytest.param('triton', marks=pytest.mark.gpu),