Skip to content

Commit

Permalink
Skip bert recipe test temporarily
Browse files Browse the repository at this point in the history
Due to torchvision::nms issue.
  • Loading branch information
tfogal committed Dec 18, 2024
1 parent 2d0199e commit 91464b3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion thunder/tests/test_recipes.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:]
Expand Down

0 comments on commit 91464b3

Please sign in to comment.