423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 15:52:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source 
Former-commit-id: 412139eaa2fde98ba19e1257d21144382a59f0d6
This commit is contained in:
hiyouga 2024-06-19 03:49:23 +08:00
parent 1ca639a777
commit 235ed85b0f
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):