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 |
|