feat: Add assistant turn
This commit is contained in:
parent
bf5154f9bc
commit
4a018e706f
1 changed files with 1 additions and 0 deletions
|
@ -69,6 +69,7 @@ def display_chat_template(chat_template: str, format_template: bool = False):
|
||||||
messages=[
|
messages=[
|
||||||
{"role": "system", "content": "I am a helpful assistant."},
|
{"role": "system", "content": "I am a helpful assistant."},
|
||||||
{"role": "user", "content": "Hello!"},
|
{"role": "user", "content": "Hello!"},
|
||||||
|
{"role": "assistant", "content": "Hello! How may I assist you today?"},
|
||||||
],
|
],
|
||||||
bos_token="[BOS]",
|
bos_token="[BOS]",
|
||||||
eos_token="[EOS]",
|
eos_token="[EOS]",
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue