Further reduce memory consumption [release]

datadreamer-dev · Apr 30, 2024 · f063b59 · f063b59
1 parent 510954c
commit f063b59
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/tests/llms/test_llms.py b/src/tests/llms/test_llms.py
@@ -1062,7 +1062,10 @@ def test_hf_chat_prompt_template_and_system_prompt_from_tokenizer(
             )
             assert _chat_prompt_template_and_system_prompt_from_tokenizer(
                 "Qwen/Qwen1.5-0.5B-Chat"
-            ) == (CHAT_PROMPT_TEMPLATES["chatml_system"], "You are a helpful assistant")
+            ) == (
+                CHAT_PROMPT_TEMPLATES["chatml_system"],
+                "You are a helpful assistant.",
+            )
             assert _chat_prompt_template_and_system_prompt_from_tokenizer(
                 "CohereForAI/c4ai-command-r-v01"
             ) == (CHAT_PROMPT_TEMPLATES["command_r"], None)