WisdomShell
/

CodeShell-7B

Text Generation

Model card Files Files and versions Community

ruixie commited on Oct 20, 2023

Commit

8cf8467

•

1 Parent(s): ae7085f

Update modeling_codeshell.py

Files changed (1) hide show

modeling_codeshell.py +2 -2

modeling_codeshell.py CHANGED Viewed

@@ -947,8 +947,8 @@ class CodeShellForCausalLM(CodeShellPreTrainedModel):
     def build_chat_input(self, query, history, tokenizer, max_new_tokens=None):
-        user_name = "\n## human:"
-        ai_name = "\n## assistant: "
         stop = '|<end>|'
         prompt = ''

     def build_chat_input(self, query, history, tokenizer, max_new_tokens=None):
+        user_name = "## human:"
+        ai_name = "## assistant: "
         stop = '|<end>|'
         prompt = ''