Skip to content

GitLab

Unverified Commit 86e70591 authored by lvhan028's avatar lvhan028 Committed by GitHub
Browse files

miss <bos> of InternLM chat template (#112)

parent c4d061d8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
lmdeploy/model.py
View file @ 86e70591
...@@ -58,8 +58,7 @@ class InternLM: ...@@ -58,8 +58,7 @@ class InternLM:
str: the concatenated prompt str: the concatenated prompt
""" """
if sequence_start: if sequence_start:
return f'{self.system}\n' \ return f'<bos>{self.user}:{prompt}{self.eoh}\n' \
f'{self.user}:{prompt}{self.eoh}\n' \
f'{self.assistant}:' f'{self.assistant}:'
else: else:
return f'\n{self.user}:{prompt}{self.eoh}\n' \ return f'\n{self.user}:{prompt}{self.eoh}\n' \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment