From 27764e3e4751c5a8095c55f1d800a3e90effd2dd Mon Sep 17 00:00:00 2001
From: Joshua Turner <jturner@handshape.com>
Date: Mon, 12 May 2025 17:42:22 -0400
Subject: [PATCH] Add support for jinja break for Cohere

---
 llama_cpp/llama_chat_format.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/llama_cpp/llama_chat_format.py b/llama_cpp/llama_chat_format.py
index 17575c700..edc644756 100644
--- a/llama_cpp/llama_chat_format.py
+++ b/llama_cpp/llama_chat_format.py
@@ -211,6 +211,7 @@ def __init__(
             loader=jinja2.BaseLoader(),
             trim_blocks=True,
             lstrip_blocks=True,
+            extensions=["jinja2.ext.loopcontrols"]
         ).from_string(self.template)
 
     def __call__(