Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -98,14 +98,14 @@ def get_inference_prompt(messages, model_name):
|
|
| 98 |
if message["role"] == "user" :
|
| 99 |
prompt += f"<|user|>\n{message['content']}<|endoftext|>\n<|assistant|>"
|
| 100 |
else : # message["role"] == "assistant"
|
| 101 |
-
prompt += f"{message['content']}<|endoftext|>\n"
|
| 102 |
elif model_name == "starChat-alpha" :
|
| 103 |
prompt = "<|system|>\n<|end|>\n"
|
| 104 |
for message in messages :
|
| 105 |
if message["role"] == "user" :
|
| 106 |
prompt += f"<|user|>\n{message['content']}<|end|>\n<|assistant|>"
|
| 107 |
else : # message["role"] == "assistant"
|
| 108 |
-
prompt += f"{message['content']}<|end|>\n"
|
| 109 |
else : # starCoder-gradio
|
| 110 |
prompt = ""
|
| 111 |
for message in messages :
|
|
|
|
| 98 |
if message["role"] == "user" :
|
| 99 |
prompt += f"<|user|>\n{message['content']}<|endoftext|>\n<|assistant|>"
|
| 100 |
else : # message["role"] == "assistant"
|
| 101 |
+
prompt += f"\n{message['content']}<|endoftext|>\n"
|
| 102 |
elif model_name == "starChat-alpha" :
|
| 103 |
prompt = "<|system|>\n<|end|>\n"
|
| 104 |
for message in messages :
|
| 105 |
if message["role"] == "user" :
|
| 106 |
prompt += f"<|user|>\n{message['content']}<|end|>\n<|assistant|>"
|
| 107 |
else : # message["role"] == "assistant"
|
| 108 |
+
prompt += f"\n{message['content']}<|end|>\n"
|
| 109 |
else : # starCoder-gradio
|
| 110 |
prompt = ""
|
| 111 |
for message in messages :
|