Consolidate Llama model usage in tests (#13094)

2025-02-14 06:18:03 +00:00
parent 40932d7a05
commit f2b20fe491
22 changed files with 45 additions and 53 deletions
--- a/tests/basic_correctness/test_cumem.py
+++ b/tests/basic_correctness/test_cumem.py
@@ -118,7 +118,7 @@ def test_cumem_with_cudagraph():
@pytest.mark.parametrize(
    "model",
    [
-        "meta-llama/Llama-3.2-1B",  # sleep mode with safetensors
+        "meta-llama/Llama-3.2-1B-Instruct",  # sleep mode with safetensors
        "facebook/opt-125m"  # sleep mode with pytorch checkpoint
    ])
 def test_end_to_end(model):