Spaces:
Sleeping
Sleeping
nroggendorff
commited on
Commit
•
e6142fd
1
Parent(s):
7e0dfdf
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ def generate_chatml(messages):
|
|
5 |
chatml = ""
|
6 |
for i, message in enumerate(messages):
|
7 |
if message.strip():
|
8 |
-
role = "<|im_start|>user
|
9 |
chatml += f"{role}\n{message.strip()}\n<|im_end|>\n"
|
10 |
return chatml.strip()
|
11 |
|
|
|
5 |
chatml = ""
|
6 |
for i, message in enumerate(messages):
|
7 |
if message.strip():
|
8 |
+
role = "<|im_start|>user" if i % 2 == 0 else "<|im_start|>assistant"
|
9 |
chatml += f"{role}\n{message.strip()}\n<|im_end|>\n"
|
10 |
return chatml.strip()
|
11 |
|