Remove gemma test

intel · Feb 28, 2024 · ccbf89c · ccbf89c
1 parent 3f47402
commit ccbf89c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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(