fix labels

2026-02-19 23:44:46 -05:00
parent 9722aaf7d8
commit eb59070040
1 changed files with 1 additions and 1 deletions
--- a/tests/prompt_strategies/test_chat_templates.py
+++ b/tests/prompt_strategies/test_chat_templates.py
@@ -159,7 +159,7 @@ class TestAssistantChatTemplateLlama3:
            -100,  # user
            -100, -100, -100,  # user eot
            -100,  # assistant
-            1781, 26966, 32007,  # assistant eot
+            16773, 26966, 32007,  # assistant eot
        ]
        # fmt: on
        LOG.debug(f"Expected input_ids: {expected_input_ids}")