423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-22 22:02:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source 
Former-commit-id: 6d2bf216ac3a48450e861148ce664dad717fd019
This commit is contained in:
hiyouga 2024-06-19 03:49:23 +08:00
parent e73a235a38
commit c9557241f6
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 system = system or self.default_system
encoded_messages = [] encoded_messages = []
for i, message in enumerate(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): if i == 0 and (system or tools):
tool_text = self.format_tools.apply(content=tools)[0] if tools else "" 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 system = system or self.default_system
encoded_messages = [] encoded_messages = []
for i, message in enumerate(messages): for i, message in enumerate(messages):
elements = self.format_prefix.apply() elements = []
if i == 0:
elements += self.format_prefix.apply()
system_text = "" system_text = ""
if i == 0 and (system or tools): if i == 0 and (system or tools):