diff --git a/thunder/tests/test_networks.py b/thunder/tests/test_networks.py index 8a42d31074..d122b00df8 100644 --- a/thunder/tests/test_networks.py +++ b/thunder/tests/test_networks.py @@ -367,10 +367,6 @@ def test_quantization(): assert_close(v, sd2[k]) -@pytest.mark.skipif( - version_between(torch.__version__, min_ver="2.6.0dev0", max_ver="2.6.0a99"), - reason="https://github.com/Lightning-AI/lightning-thunder/issues/1471", -) @thunder.tests.framework.requiresCUDA def test_thunderfx_mistral_nemo_small(): """ @@ -422,10 +418,6 @@ def test_thunderfx_mistral_nemo_small(): # disabled "Qwen/Qwen2.5-7B-Instruct" see https://github.com/NVIDIA/Fuser/issues/3682 -@pytest.mark.skipif( - version_between(torch.__version__, min_ver="2.6.0dev0", max_ver="2.6.0a99"), - reason="https://github.com/Lightning-AI/lightning-thunder/issues/1471", -) @thunder.tests.framework.requiresCUDA @pytest.mark.parametrize("model_id", ["microsoft/Phi-3-mini-128k-instruct"]) def test_hf_for_nemo(model_id): diff --git a/thunder/tests/test_recipes.py b/thunder/tests/test_recipes.py index e5c62138da..826df95171 100644 --- a/thunder/tests/test_recipes.py +++ b/thunder/tests/test_recipes.py @@ -24,10 +24,6 @@ def test_recipe_basic_bert(): assert_close(actual, expected) -@pytest.mark.skipif( - version_between(torch.__version__, min_ver="2.6.0dev0", max_ver="2.6.0a99"), - reason="https://github.com/Lightning-AI/lightning-thunder/issues/1471", -) def test_recipe_basic_bert_dynamo(): bert = transformers.BertForSequenceClassification(transformers.BertConfig()) del bert.bert.encoder.layer[1:]