kyy/llm_trainer
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix jinja template

Browse Source
This commit is contained in:
hiyouga committed 2024-06-19 20:03:50 +08:00
1 parent 4cff6a4ad5
commit 2b596fb55f
3 files changed
+46 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/llamafactory/data/template.py
+5 -1
View File
@@ -338,7 +338,11 @@ def _convert_slots_to_jinja(slots: "SLOTS", tokenizer: "PreTrainedTokenizer", pl
def _get_jinja_template(template: "Template", tokenizer: "PreTrainedTokenizer") -> str:
jinja_template = _convert_slots_to_jinja(template.format_prefix.apply(), tokenizer)
jinja_template = ""
prefix = _convert_slots_to_jinja(template.format_prefix.apply(), tokenizer)
if prefix:
jinja_template += "{{ " + prefix + " }}"
if template.default_system:
jinja_template += "{% set system_message = '" + _jinja_escape(template.default_system) + "' %}"