Update run.py
Browse files
run.py
CHANGED
|
@@ -87,13 +87,15 @@ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
|
|
| 87 |
|
| 88 |
|
| 89 |
def format_prompt(message, history):
|
| 90 |
-
|
| 91 |
-
|
| 92 |
prompt = "" #"<s>"
|
| 93 |
for user_prompt, bot_response in history:
|
| 94 |
prompt += f"[INST] {user_prompt} [/INST]"
|
| 95 |
prompt += f" {bot_response}</s> "
|
| 96 |
prompt += f"[INST] {message} [/INST]"
|
|
|
|
|
|
|
| 97 |
return prompt
|
| 98 |
|
| 99 |
def response(
|
|
|
|
| 87 |
|
| 88 |
|
| 89 |
def format_prompt(message, history):
|
| 90 |
+
print("HOSTORY")
|
| 91 |
+
print(history)
|
| 92 |
prompt = "" #"<s>"
|
| 93 |
for user_prompt, bot_response in history:
|
| 94 |
prompt += f"[INST] {user_prompt} [/INST]"
|
| 95 |
prompt += f" {bot_response}</s> "
|
| 96 |
prompt += f"[INST] {message} [/INST]"
|
| 97 |
+
print("Final P")
|
| 98 |
+
print(prompt)
|
| 99 |
return prompt
|
| 100 |
|
| 101 |
def response(
|