add extra test back

mosaicml · Nov 16, 2023 · 2f8afa2 · 2f8afa2
1 parent 43c4195
commit 2f8afa2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_mpt_gen.py b/tests/test_mpt_gen.py
@@ -145,6 +145,7 @@ def test_mpt_generate_callback(attn_impl: str, use_alibi: bool,
 
 @pytest.mark.gpu
 @pytest.mark.parametrize('attn_impl', ['triton', 'torch'])
+@pytest.mark.parametrize('use_alibi', [True, False])
 def test_mpt_generate_callback_not_tied(
         attn_impl: str, build_tiny_mpt: Callable[..., ComposerMPTCausalLM],
         tiny_ft_dataloader: DataLoader):