From eb590700402a1c0901ea19a7d2d78cd7637b20d9 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 19 Feb 2026 23:44:46 -0500 Subject: [PATCH] fix labels --- tests/prompt_strategies/test_chat_templates.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/prompt_strategies/test_chat_templates.py b/tests/prompt_strategies/test_chat_templates.py index f5d9d6f54..21d6efd3a 100644 --- 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}")