From d71a62e98e4bee7f72f878089288d629786abbdc Mon Sep 17 00:00:00 2001 From: grencez Date: Thu, 25 Jul 2024 03:34:55 -0700 Subject: [PATCH] but really tho --- example/prompt/assistant_llama/setting.sxpb | 1 + test/example/prompt/CMakeLists.txt | 2 ++ 2 files changed, 3 insertions(+) diff --git a/example/prompt/assistant_llama/setting.sxpb b/example/prompt/assistant_llama/setting.sxpb index 4417de1..24fb783 100644 --- a/example/prompt/assistant_llama/setting.sxpb +++ b/example/prompt/assistant_llama/setting.sxpb @@ -2,6 +2,7 @@ (m (prefix "<|start_header_id|>user<|end_header_id|>\n\n") (suffix "<|eot_id|>\n") + ) (m (prefix "<|start_header_id|>assistant<|end_header_id|>\n\n") (suffix "<|eot_id|>\n") diff --git a/test/example/prompt/CMakeLists.txt b/test/example/prompt/CMakeLists.txt index d065d78..6c5ae5b 100644 --- a/test/example/prompt/CMakeLists.txt +++ b/test/example/prompt/CMakeLists.txt @@ -17,6 +17,8 @@ add_test(NAME example_prompt_assistant_coprocess_parse_test COMMAND example_prom "${PROJECT_SOURCE_DIR}/example/prompt/assistant_coprocess/setting.sxpb") add_test(NAME example_prompt_assistant_gemma_parse_test COMMAND example_prompt_parse_test "${PROJECT_SOURCE_DIR}/example/prompt/assistant_gemma/setting.sxpb") +add_test(NAME example_prompt_assistant_llama_parse_test COMMAND example_prompt_parse_test + "${PROJECT_SOURCE_DIR}/example/prompt/assistant_llama/setting.sxpb") add_test(NAME example_prompt_assistant_mistral_parse_test COMMAND example_prompt_parse_test "${PROJECT_SOURCE_DIR}/example/prompt/assistant_mistral/setting.sxpb") add_test(NAME example_prompt_assistant_plain_parse_test COMMAND example_prompt_parse_test