Spaces:
Sleeping
Sleeping
Commit
·
ddba001
1
Parent(s):
457a722
debug
Browse files
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 |
-
|
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()
|