!205 [bugfix] fix unmatched arguments

Merge pull request !205 from Calvin Huang/master
2025-05-12 11:51:28 +00:00
parent 9f25c83026
commit 6c27012844
1 changed files with 1 additions and 1 deletions
--- a/src/openmind/flow/datasets/template.py
+++ b/src/openmind/flow/datasets/template.py
@ -130,7 +130,7 @@ class Template:
        r"""
        Returns a single pair of token ids representing prompt and response respectively.
        """
-        encoded_messages = self.encode(tokenizer, messages, system, tools)
+        encoded_messages = self.encode(tokenizer, messages, system)
        prompt_ids = []
        for encoded_ids in encoded_messages[:-1]:
            prompt_ids += encoded_ids