Spaces:
Runtime error
Runtime error
Aleksandr Maiorov
commited on
Commit
·
9ab8f00
1
Parent(s):
58d7ca2
v 0.1
Browse files- добавлен системный промт;
- add_generation_prompt=False
app.py
CHANGED
@@ -14,7 +14,7 @@ logger.info("Запускаемся... 🥳🥳🥳")
|
|
14 |
|
15 |
app = FastAPI()
|
16 |
model_url = "https://huggingface.co/Qwen/Qwen2.5-7B-Instruct-GGUF/resolve/main/qwen2.5-7b-instruct-q3_k_m.gguf"
|
17 |
-
|
18 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
|
19 |
|
20 |
|
@@ -27,9 +27,12 @@ def messages_to_prompt(messages):
|
|
27 |
|
28 |
|
29 |
def completion_to_prompt(completion):
|
30 |
-
messages = [
|
|
|
|
|
|
|
31 |
prompt = tokenizer.apply_chat_template(
|
32 |
-
messages, tokenize=False, add_generation_prompt=
|
33 |
)
|
34 |
return prompt
|
35 |
|
|
|
14 |
|
15 |
app = FastAPI()
|
16 |
model_url = "https://huggingface.co/Qwen/Qwen2.5-7B-Instruct-GGUF/resolve/main/qwen2.5-7b-instruct-q3_k_m.gguf"
|
17 |
+
SYSTEM_PROMPT = ''
|
18 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-7B-Instruct")
|
19 |
|
20 |
|
|
|
27 |
|
28 |
|
29 |
def completion_to_prompt(completion):
|
30 |
+
messages = [
|
31 |
+
{"role": "system", "content": SYSTEM_PROMPT or "Ответ должен быть точным, кратким и с юмором."},
|
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 |
|