Spaces:
Runtime error
Runtime error
gorkemgoknar
commited on
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 |
)
|