423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-17 04:10:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor constants

Browse Source
This commit is contained in:
hiyouga
2023-11-10 14:16:10 +08:00
parent 415bca900e
commit 3697a3dc9a
6 changed files with 243 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -447,6 +447,25 @@ register_template(
)
r"""
Supports: https://huggingface.co/tiiuae/falcon-180B-chat
"""
register_template(
name="falcon",
prefix=[
"{{system}}"
],
prompt=[
"User: {{query}}\nFalcon:"
],
system="",
sep=[
"\n"
],
efficient_eos=True
)
r"""
Supports: https://huggingface.co/internlm/internlm-chat-7b
https://huggingface.co/internlm/internlm-chat-20b