Spaces:
Runtime error
Runtime error
Update main.py
Browse files
main.py
CHANGED
@@ -26,8 +26,8 @@ max_new_tokens = 2048
|
|
26 |
|
27 |
def generater(message, history, temperature, top_p, top_k):
|
28 |
prompt = "<s>"
|
|
|
29 |
for user_message, assistant_message in history:
|
30 |
-
prompt += model.config["systemPrompt"]
|
31 |
prompt += model.config["promptTemplate"].format(user_message)
|
32 |
prompt += assistant_message + "</s>"
|
33 |
prompt += model.config["promptTemplate"].format(message)
|
|
|
26 |
|
27 |
def generater(message, history, temperature, top_p, top_k):
|
28 |
prompt = "<s>"
|
29 |
+
prompt += model.config["systemPrompt"]
|
30 |
for user_message, assistant_message in history:
|
|
|
31 |
prompt += model.config["promptTemplate"].format(user_message)
|
32 |
prompt += assistant_message + "</s>"
|
33 |
prompt += model.config["promptTemplate"].format(message)
|