We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a7da023 commit 6066a01Copy full SHA for 6066a01
mellea/backends/huggingface.py
@@ -341,6 +341,7 @@ def _generate_from_context_standard(
341
input_ids = self._tokenizer.apply_chat_template( # type: ignore
342
ctx_as_conversation,
343
tools=convert_tools_to_json(tools), # type: ignore
344
+ add_generation_prompt=True,
345
return_tensors="pt",
346
**self._make_backend_specific_and_remove(model_options),
347
).to(self._device) # type: ignore
0 commit comments