Cannot fetch model from github action

intel · Feb 28, 2024 · d00c83d · d00c83d
1 parent 1bee658
commit d00c83d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/python/test_optimizations.py b/test/python/test_optimizations.py
@@ -110,6 +110,9 @@ def get_model(model_name, hidden_size, intermediate_size, bias):
 @pytest.mark.parametrize("bias", [True, False])
 def test_fusion(model_name, hidden_size, intermediate_size, batch, bias):
 
+    if model_name == "GemmaMLP":
+        pytest.skip("Cannot fetch model from github action")
+
     model = get_model(model_name, hidden_size, intermediate_size, bias)
     example_input = torch.rand((batch, hidden_size)) - 0.5