diff --git a/mlx_lm/server.py b/mlx_lm/server.py
index 02b3d9c..712e4e1 100644
--- a/mlx_lm/server.py
+++ b/mlx_lm/server.py
@@ -653,7 +653,7 @@ class APIHandler(BaseHTTPRequestHandler):
         ):
             segment = gen_response.text
             text += segment
-            logging.debug(text)
+            logging.debug(segment)
             token = gen_response.token
             logprobs = gen_response.logprobs
             tokens.append(token)
diff --git a/mlx_lm/utils.py b/mlx_lm/utils.py
index 5305d2e..91935ed 100644
--- a/mlx_lm/utils.py
+++ b/mlx_lm/utils.py
@@ -112,6 +112,7 @@ def get_model_path(path_or_hf_repo: str, revision: Optional[str] = None) -> Path
                     "tiktoken.model",
                     "*.txt",
                     "*.jsonl",
+                    "*.jinja",
                 ],
             )
         )