Skip to content

Commit 5922437

Browse files
Update huggingface_utils.py
1 parent c16a69d commit 5922437

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/agentlab/llm/huggingface_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from transformers import AutoTokenizer, GPT2TokenizerFast
77

88
from agentlab.llm.base_api import AbstractChatModel
9+
from agentlab.llm.llm_utils import Discussion
910
from agentlab.llm.prompt_templates import PromptTemplate, get_prompt_template
1011

1112

@@ -59,7 +60,8 @@ def __call__(
5960
if self.tokenizer:
6061
# messages_formated = _convert_messages_to_dict(messages) ## ?
6162
try:
62-
messages.merge()
63+
if isinstance(messages, Discussion):
64+
messages.merge()
6365
prompt = self.tokenizer.apply_chat_template(messages, tokenize=False)
6466
except Exception as e:
6567
if "Conversation roles must alternate" in str(e):

0 commit comments

Comments
 (0)