fix

mosaicml · Nov 18, 2023 · 3a863ca · 3a863ca
1 parent 1ba798e
commit 3a863ca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -304,6 +304,7 @@ def test_full_forward_and_backward_t5_small(batch_size: int = 2):
     assert not torch.equal(original_params, updated_params)
 
 
+@pytest.mark.gpu
 @pytest.mark.parametrize(
     'attn_impl,precision',
     [('torch', torch.float16), ('torch', torch.bfloat16),