Spaces:
Sleeping
Sleeping
Redone system prompt
Browse files
app.py
CHANGED
|
@@ -17,8 +17,8 @@ def format_prompt(message, history, systemPrompt):
|
|
| 17 |
# First everything is converted into role format
|
| 18 |
# First a system prompt
|
| 19 |
messages = []
|
| 20 |
-
|
| 21 |
-
|
| 22 |
# Followed by the message history
|
| 23 |
for user_message, bot_message in history:
|
| 24 |
messages.append({"role": "user", "content": user_message})
|
|
@@ -27,8 +27,8 @@ def format_prompt(message, history, systemPrompt):
|
|
| 27 |
messages.append({"role": "user", "content": message})
|
| 28 |
# The tokenizer converts into the model format
|
| 29 |
messages = tokenizer.apply_chat_template(messages, tokenize=False)
|
| 30 |
-
systemPromptMessage = f"[INST]{systemPrompt}[/INST]"
|
| 31 |
-
messages=systemPromptMessage+messages
|
| 32 |
return messages
|
| 33 |
|
| 34 |
|
|
|
|
| 17 |
# First everything is converted into role format
|
| 18 |
# First a system prompt
|
| 19 |
messages = []
|
| 20 |
+
messages.append({"role": "user", "content": systemPrompt})
|
| 21 |
+
messages.append({"role": "assistant", "content": ""})
|
| 22 |
# Followed by the message history
|
| 23 |
for user_message, bot_message in history:
|
| 24 |
messages.append({"role": "user", "content": user_message})
|
|
|
|
| 27 |
messages.append({"role": "user", "content": message})
|
| 28 |
# The tokenizer converts into the model format
|
| 29 |
messages = tokenizer.apply_chat_template(messages, tokenize=False)
|
| 30 |
+
#systemPromptMessage = f"[INST]{systemPrompt}[/INST]"
|
| 31 |
+
#messages=systemPromptMessage+messages
|
| 32 |
return messages
|
| 33 |
|
| 34 |
|