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 c16a69d commit 5922437Copy full SHA for 5922437
src/agentlab/llm/huggingface_utils.py
@@ -6,6 +6,7 @@
6
from transformers import AutoTokenizer, GPT2TokenizerFast
7
8
from agentlab.llm.base_api import AbstractChatModel
9
+from agentlab.llm.llm_utils import Discussion
10
from agentlab.llm.prompt_templates import PromptTemplate, get_prompt_template
11
12
@@ -59,7 +60,8 @@ def __call__(
59
60
if self.tokenizer:
61
# messages_formated = _convert_messages_to_dict(messages) ## ?
62
try:
- messages.merge()
63
+ if isinstance(messages, Discussion):
64
+ messages.merge()
65
prompt = self.tokenizer.apply_chat_template(messages, tokenize=False)
66
except Exception as e:
67
if "Conversation roles must alternate" in str(e):
0 commit comments