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 :
|