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

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

Files changed (1) hide show
  1. app.py +2 -2
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=False
25
  )
26
  return prompt
27
 
@@ -32,7 +32,7 @@ def completion_to_prompt(completion):
32
  {"role": "user", "content": completion},
33
  ]
34
  prompt = tokenizer.apply_chat_template(
35
- messages, tokenize=False, add_generation_prompt=False
36
  )
37
  return prompt
38
 
 
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
 
 
32
  {"role": "user", "content": completion},
33
  ]
34
  prompt = tokenizer.apply_chat_template(
35
+ messages, tokenize=False, add_generation_prompt=True
36
  )
37
  return prompt
38