Aleksandr Maiorov commited on
Commit
1366f84
·
1 Parent(s): 9ab8f00

- добавлен системный промт;
- add_generation_prompt=False

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
21
  def messages_to_prompt(messages):
22
  messages = [{"role": m.role.value, "content": m.content} for m in messages]
23
  prompt = tokenizer.apply_chat_template(
24
- messages, tokenize=False, add_generation_prompt=True
25
  )
26
  return prompt
27
 
 
21
  def messages_to_prompt(messages):
22
  messages = [{"role": m.role.value, "content": m.content} for m in messages]
23
  prompt = tokenizer.apply_chat_template(
24
+ messages, tokenize=False, add_generation_prompt=False
25
  )
26
  return prompt
27