From dec05878afad039e7d8585daf337f3d3097cdb8b Mon Sep 17 00:00:00 2001 From: Michael Benayoun Date: Thu, 12 Dec 2024 15:33:46 +0100 Subject: [PATCH] Use ungated models for the tests --- tests/executorch/export/test_exporters_executorch.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/executorch/export/test_exporters_executorch.py b/tests/executorch/export/test_exporters_executorch.py index a4521bc018..f2467105e4 100644 --- a/tests/executorch/export/test_exporters_executorch.py +++ b/tests/executorch/export/test_exporters_executorch.py @@ -33,7 +33,7 @@ def test_helps_no_raise(self): @slow @pytest.mark.run_slow def test_llama3_2_1b_export_to_executorch(self): - model_id = "meta-llama/Llama-3.2-1B" + model_id = "NousResearch/Llama-3.2-1B" task = "text-generation" recipe = "xnnpack" with tempfile.TemporaryDirectory() as tempdir: @@ -47,7 +47,7 @@ def test_llama3_2_1b_export_to_executorch(self): @slow @pytest.mark.run_slow def test_llama3_2_3b_export_to_executorch(self): - model_id = "meta-llama/Llama-3.2-3B" + model_id = "NousResearch/Hermes-3-Llama-3.2-3B" task = "text-generation" recipe = "xnnpack" with tempfile.TemporaryDirectory() as tempdir: @@ -75,7 +75,7 @@ def test_qwen2_5_export_to_executorch(self): @slow @pytest.mark.run_slow def test_gemma2_export_to_executorch(self): - model_id = "google/gemma-2-2b" + model_id = "unsloth/gemma-2-2b-it" task = "text-generation" recipe = "xnnpack" with tempfile.TemporaryDirectory() as tempdir: @@ -89,7 +89,7 @@ def test_gemma2_export_to_executorch(self): @slow @pytest.mark.run_slow def test_gemma_export_to_executorch(self): - model_id = "google/gemma-2b" + model_id = "weqweasdas/RM-Gemma-2B" task = "text-generation" recipe = "xnnpack" with tempfile.TemporaryDirectory() as tempdir: