KaiChen1998 commited on
Commit
7591e86
·
verified ·
1 Parent(s): 0224173

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -204,7 +204,7 @@ def http_bot(state, temperature, top_p, max_new_tokens, speaker):
204
  # First round of conversation
205
  if "llama3" in model_name.lower():
206
  template_name = 'llama3_demo'
207
- elif "qwen2" in model_name.lower():
208
  template_name = 'qwen2_demo'
209
  else:
210
  template_name = "default"
 
204
  # First round of conversation
205
  if "llama3" in model_name.lower():
206
  template_name = 'llama3_demo'
207
+ elif "qwen-2" in model_name.lower():
208
  template_name = 'qwen2_demo'
209
  else:
210
  template_name = "default"