update UI, fix #212

Former-commit-id: ac92c2bd7c47353759474fad9412f21b38c65501
2023-07-20 22:09:06 +08:00
parent 64db4abc68
commit 9b3304b054
12 changed files with 155 additions and 45 deletions
--- a/src/llmtuner/extras/constants.py
+++ b/src/llmtuner/extras/constants.py
@@ -36,7 +36,7 @@ SUPPORTED_MODELS = {
    "InternLM-7B-Chat": "internlm/internlm-chat-7b"
 }

-DEFAULT_MODULE = { # will be deprecated
+DEFAULT_MODULE = {
    "LLaMA": "q_proj,v_proj",
    "LLaMA2": "q_proj,v_proj",
    "BLOOM": "query_key_value",
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@@ -215,7 +215,7 @@ Supports: https://huggingface.co/baichuan-inc/Baichuan-13B-Chat
 register_template(
    name="baichuan",
    prefix="",
-    prompt=" <reserved_102> {query} <reserved_103> ",
+    prompt="<reserved_102>{query}<reserved_103>",
    sep="</s>",
    use_history=True
 )