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

fix #2629

Browse Source 
Former-commit-id: c18822669568327d4fbf480a80c5fe5b8fc95e7a
This commit is contained in:
hiyouga
2024-02-29 00:37:29 +08:00
parent 544e7a491b
commit 0e621c2dc9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/data/template.py
View File

@@ -258,7 +258,7 @@ def get_template_and_fix_tokenizer(
template = templates["vanilla"] # placeholder template = templates["vanilla"] # placeholder
else: else:
template = templates.get(name, None) template = templates.get(name, None)
if templates is None: if template is None:
raise ValueError("Template {} does not exist.".format(name)) raise ValueError("Template {} does not exist.".format(name))
stop_words = template.stop_words stop_words = template.stop_words