diff --git a/thunder/tests/test_recipes.py b/thunder/tests/test_recipes.py index e5c62138da..f5f8c40b2c 100644 --- a/thunder/tests/test_recipes.py +++ b/thunder/tests/test_recipes.py @@ -6,7 +6,7 @@ from torch.testing import assert_close, make_tensor from thunder.tests.framework import version_between - +@pytest.mark.skip(reason="torchvision::nms issue with transformers") def test_recipe_basic_bert(): bert = transformers.BertForSequenceClassification(transformers.BertConfig()) del bert.bert.encoder.layer[1:]