Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ def respond(
|
|
| 29 |
top_p,
|
| 30 |
):
|
| 31 |
stop_tokens = ["</s>", "[INST]", "[INST] ", "<s>", "[/INST]", "[/INST] "]
|
| 32 |
-
chat_template = '<s>[INST] ' +
|
| 33 |
for human, assistant in history:
|
| 34 |
chat_template += human + ' [/INST] ' + assistant + '</s>[INST]'
|
| 35 |
chat_template += ' ' + message + ' [/INST]'
|
|
|
|
| 29 |
top_p,
|
| 30 |
):
|
| 31 |
stop_tokens = ["</s>", "[INST]", "[INST] ", "<s>", "[/INST]", "[/INST] "]
|
| 32 |
+
chat_template = '<s>[INST] ' + system_message
|
| 33 |
for human, assistant in history:
|
| 34 |
chat_template += human + ' [/INST] ' + assistant + '</s>[INST]'
|
| 35 |
chat_template += ' ' + message + ' [/INST]'
|