diff --git a/test/python/test_optimizations.py b/test/python/test_optimizations.py index 691c454..68dc829 100644 --- a/test/python/test_optimizations.py +++ b/test/python/test_optimizations.py @@ -132,6 +132,9 @@ def test_fusion(model_name, hidden_size, intermediate_size, batch, bias): @pytest.mark.parametrize("bias", [True, False]) def test_model(model_name, hidden_size, intermediate_size, sequence_length, bias): + if model_name == "GemmaMLP": + pytest.skip("Cannot fetch model from github action") + with torch.no_grad(): model = get_model(model_name, hidden_size, intermediate_size, bias).eval() example_input = torch.randint(