Spaces:
Runtime error
Runtime error
Aleksandr Maiorov
commited on
Commit
·
1b4738b
1
Parent(s):
1366f84
v 0.1
Browse files- добавлен системный промт;
- add_generation_prompt=False
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=
|
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=
|
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 |
|