adnan5253
diff --git a/Collapse file
‎llama_cpp/llama_chat_format.py‎
Copy file name to clipboardExpand all lines: llama_cpp/llama_chat_format.py
+2-1Lines changed: 2 additions & 1 deletion b/Collapse file
‎llama_cpp/llama_chat_format.py‎
Copy file name to clipboardExpand all lines: llama_cpp/llama_chat_format.py
+2-1Lines changed: 2 additions & 1 deletion
@@ -11,6 +11,7 @@
 from typing import Any, Dict, Iterator, List, Literal, Optional, Tuple, Union, Protocol, cast
 
 import jinja2
+from jinja2.sandbox import ImmutableSandboxedEnvironment
 
 import numpy as np
 import numpy.typing as npt
@@ -191,7 +192,7 @@ def __init__(
         self.add_generation_prompt = add_generation_prompt
         self.stop_token_ids = set(stop_token_ids) if stop_token_ids is not None else None
 
-        self._environment = jinja2.Environment(
+        self._environment = ImmutableSandboxedEnvironment(
             loader=jinja2.BaseLoader(),
             trim_blocks=True,
             lstrip_blocks=True,