yentinglin commited on
Commit
9077757
1 Parent(s): 9ee12c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("vicuna").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:
 
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: