fix chat templates (#205)

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)

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",
                 ],
             )
         )