Test if it OOMs

huggingface · Dec 12, 2024 · 32d7d79 · 32d7d79
1 parent dec0587
commit 32d7d79
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/executorch/export/test_exporters_executorch.py b/tests/executorch/export/test_exporters_executorch.py
@@ -74,6 +74,7 @@ def test_qwen2_5_export_to_executorch(self):
 
     @slow
     @pytest.mark.run_slow
+    @pytest.mark.skip
     def test_gemma2_export_to_executorch(self):
         model_id = "unsloth/gemma-2-2b-it"
         task = "text-generation"
@@ -88,6 +89,7 @@ def test_gemma2_export_to_executorch(self):
 
     @slow
     @pytest.mark.run_slow
+    @pytest.mark.skip
     def test_gemma_export_to_executorch(self):
         model_id = "weqweasdas/RM-Gemma-2B"
         task = "text-generation"