laurenssam commited on
Commit
ddba001
·
1 Parent(s): 457a722
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -201,6 +201,10 @@ def http_bot(state, temperature, top_p, max_new_tokens):
201
  if len(state.messages) == state.offset + 2:
202
  # First round of conversation
203
  template_name = 'phi'
 
 
 
 
204
  # if "tinyllava" in model_name.lower():
205
  # if "3.1b" in model_name.lower() or "phi" in model_name.lower():
206
  # template_name = "phi"
@@ -243,10 +247,7 @@ def http_bot(state, temperature, top_p, max_new_tokens):
243
  # template_name = "llama_2"
244
  # else:
245
  # template_name = "vicuna_v1"
246
- new_state = conv_templates[template_name].copy()
247
- new_state.append_message(new_state.roles[0], state.messages[-2][1])
248
- new_state.append_message(new_state.roles[1], None)
249
- state = new_state
250
 
251
  # Construct prompt
252
  prompt = state.get_prompt()
 
201
  if len(state.messages) == state.offset + 2:
202
  # First round of conversation
203
  template_name = 'phi'
204
+ new_state = conv_templates[template_name].copy()
205
+ new_state.append_message(new_state.roles[0], state.messages[-2][1])
206
+ new_state.append_message(new_state.roles[1], None)
207
+ state = new_state
208
  # if "tinyllava" in model_name.lower():
209
  # if "3.1b" in model_name.lower() or "phi" in model_name.lower():
210
  # template_name = "phi"
 
247
  # template_name = "llama_2"
248
  # else:
249
  # template_name = "vicuna_v1"
250
+
 
 
 
251
 
252
  # Construct prompt
253
  prompt = state.get_prompt()