Commit
·
cce82f4
1
Parent(s):
0e37210
Update app.py
Browse files
app.py
CHANGED
|
@@ -153,7 +153,7 @@ llm = Llama(model_path=model_path,n_gpu_layers=GPU_LAYERS,max_new_tokens=256, co
|
|
| 153 |
|
| 154 |
|
| 155 |
# Mistral formatter
|
| 156 |
-
def format_prompt_mistral(message, history):
|
| 157 |
prompt = (
|
| 158 |
"<s>[INST]" + system_message + "[/INST]" + system_understand_message + "</s>"
|
| 159 |
)
|
|
|
|
| 153 |
|
| 154 |
|
| 155 |
# Mistral formatter
|
| 156 |
+
def format_prompt_mistral(message, history, system_message=system_message):
|
| 157 |
prompt = (
|
| 158 |
"<s>[INST]" + system_message + "[/INST]" + system_understand_message + "</s>"
|
| 159 |
)
|