diff --git a/llamafile/chatbot_repl.cpp b/llamafile/chatbot_repl.cpp index a9aad32b51..0e8ccbdf64 100644 --- a/llamafile/chatbot_repl.cpp +++ b/llamafile/chatbot_repl.cpp @@ -129,8 +129,9 @@ void repl() { } // perform important setup - HighlightMarkdown highlighter; - ColorBleeder bleeder(&highlighter); + HighlightTxt txt; + HighlightMarkdown markdown; + ColorBleeder bleeder(is_base_model() ? (Highlight *)&txt : (Highlight *)&markdown); llama_sampling_context *sampler = llama_sampling_init(g_params.sparams); signal(SIGINT, on_sigint);