Aleksandr Maiorov commited on
Commit
9ab8f00
·
1 Parent(s): 58d7ca2

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

Files changed (1) hide show
  1. app.py +6 -3
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 = [{"role": "user", "content": completion}]
 
 
 
31
  prompt = tokenizer.apply_chat_template(
32
- messages, tokenize=False, add_generation_prompt=True
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