p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug

This commit is contained in:
hiyouga 2024-06-19 03:49:23 +08:00
parent 4f22eae8f4
commit 6d2bf216ac
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/llamafactory/data/template.py
View File

@ -102,7 +102,10 @@ class Template:
system = system or self.default_system
encoded_messages = []
for i, message in enumerate(messages):
elements = self.format_prefix.apply()
elements = []
if i == 0:
elements += self.format_prefix.apply()
if i == 0 and (system or tools):
tool_text = self.format_tools.apply(content=tools)[0] if tools else ""
@ -194,7 +197,10 @@ class Llama2Template(Template):
system = system or self.default_system
encoded_messages = []
for i, message in enumerate(messages):
elements = self.format_prefix.apply()
elements = []
if i == 0:
elements += self.format_prefix.apply()
system_text = ""
if i == 0 and (system or tools):