omkar56 commited on
Commit
e66d372
·
1 Parent(s): 9a0cf0d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -26,8 +26,8 @@ max_new_tokens = 2048
26
 
27
  def generater(message, history, temperature, top_p, top_k):
28
  prompt = "<s>"
 
29
  for user_message, assistant_message in history:
30
- prompt += model.config["systemPrompt"]
31
  prompt += model.config["promptTemplate"].format(user_message)
32
  prompt += assistant_message + "</s>"
33
  prompt += model.config["promptTemplate"].format(message)
 
26
 
27
  def generater(message, history, temperature, top_p, top_k):
28
  prompt = "<s>"
29
+ prompt += model.config["systemPrompt"]
30
  for user_message, assistant_message in history:
 
31
  prompt += model.config["promptTemplate"].format(user_message)
32
  prompt += assistant_message + "</s>"
33
  prompt += model.config["promptTemplate"].format(message)