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

fix vicuna template

Browse Source 
Former-commit-id: a98eda0803e4b73a24f12d848e14161451921e98
This commit is contained in:
hiyouga
2023-10-27 22:15:25 +08:00
parent 6f09f50d02
commit 7da2aad6ee
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/llmtuner/extras/template.py
View File

@@ -336,7 +336,7 @@ register_template(
"{{system}}"
],
prompt=[
"USER: {{query}} ASSISTANT: "
"USER: {{query}} ASSISTANT:"
],
system=(
"A chat between a curious user and an artificial intelligence assistant. "
@@ -366,6 +366,7 @@ register_template(
r"""
Supports: https://huggingface.co/IDEA-CCNL/Ziya-LLaMA-13B-v1
https://huggingface.co/IDEA-CCNL/Ziya-LLaMA-13B-v1.1
https://huggingface.co/IDEA-CCNL/Ziya2-13B-Chat
"""
register_template(
@@ -397,7 +398,7 @@ register_template(
"{{system}}"
],
prompt=[
"Human: {{query}}###Assistant: "
"Human: {{query}}###Assistant:"
],
system=(
"A chat between a curious human and an artificial intelligence assistant. "