Merge branch 'ml-explore:main' into adding-support-for-mamba2

This commit is contained in:
Gökdeniz Gülmez
2024-10-04 22:25:31 +02:00
committed by GitHub
2 changed files with 5 additions and 0 deletions

View File

@@ -594,6 +594,7 @@ class APIHandler(BaseHTTPRequestHandler):
):
prompt = self.tokenizer.apply_chat_template(
body["messages"],
body.get("tools", None),
tokenize=True,
add_generation_prompt=True,
)