Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,11 +26,10 @@ def format_prompt(message, history, systemPrompt):
|
|
| 26 |
# Followed by the current message
|
| 27 |
messages.append({"role": "user", "content": message})
|
| 28 |
# The tokenizer converts into the model format
|
| 29 |
-
conversationalPart = tokenizer.apply_chat_template(
|
| 30 |
conversationalPart = conversationalPart.removeprefix('<s>')
|
| 31 |
systemPromptMessage = f"<s>[INST] <<SYS>>\n{systemPrompt}\n<</SYS>>\n\n "
|
| 32 |
-
forPrompt=systemPromptMessage+
|
| 33 |
-
print(forPrompt)
|
| 34 |
return forPrompt
|
| 35 |
|
| 36 |
|
|
|
|
| 26 |
# Followed by the current message
|
| 27 |
messages.append({"role": "user", "content": message})
|
| 28 |
# The tokenizer converts into the model format
|
| 29 |
+
conversationalPart = tokenizer.apply_chat_template(messages, tokenize=False)
|
| 30 |
conversationalPart = conversationalPart.removeprefix('<s>')
|
| 31 |
systemPromptMessage = f"<s>[INST] <<SYS>>\n{systemPrompt}\n<</SYS>>\n\n "
|
| 32 |
+
forPrompt=systemPromptMessage+messages
|
|
|
|
| 33 |
return forPrompt
|
| 34 |
|
| 35 |
|