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

fix jinja template

Browse Source 
Former-commit-id: 0ebf2e2ee23918d28b0cbb20ba456732d6eedfbb
This commit is contained in:
hiyouga
2024-06-19 20:03:50 +08:00
parent 3e0fa4a8da
commit c65f7e9bd5
3 changed files with 46 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/data/template.py
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) + "' %}"