From 32d7d7901de4699a681496b694d4f0fd7e59c107 Mon Sep 17 00:00:00 2001 From: Michael Benayoun Date: Thu, 12 Dec 2024 16:44:57 +0100 Subject: [PATCH] Test if it OOMs --- tests/executorch/export/test_exporters_executorch.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/executorch/export/test_exporters_executorch.py b/tests/executorch/export/test_exporters_executorch.py index f2467105e4..ddce086b80 100644 --- 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"