Spaces:
Running
Running
yentinglin
commited on
Commit
•
9077757
1
Parent(s):
9ee12c9
Update app.py
Browse files
app.py
CHANGED
@@ -141,7 +141,7 @@ with gr.Blocks() as demo:
|
|
141 |
|
142 |
|
143 |
def bot(history, max_new_tokens, temperature, top_p, top_k, system_prompt):
|
144 |
-
conv = get_default_conv_template("
|
145 |
roles = {"human": conv.roles[0], "gpt": conv.roles[1]} # map human to USER and gpt to ASSISTANT
|
146 |
conv.system = system_prompt
|
147 |
for user, bot in history:
|
|
|
141 |
|
142 |
|
143 |
def bot(history, max_new_tokens, temperature, top_p, top_k, system_prompt):
|
144 |
+
conv = get_default_conv_template("twllm_v2").copy()
|
145 |
roles = {"human": conv.roles[0], "gpt": conv.roles[1]} # map human to USER and gpt to ASSISTANT
|
146 |
conv.system = system_prompt
|
147 |
for user, bot in history:
|