gorkemgoknar commited on
Commit
cce82f4
1 Parent(s): 0e37210

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
  )