IELTS8 commited on
Commit
2bbbebc
1 Parent(s): 11bcf37

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -187,7 +187,7 @@ def http_bot(
187
 
188
  if len(state.messages) == state.offset + 2:
189
  # First round of conversation
190
- if "llava" in model_name.lower():
191
  if "llama-2" in model_name.lower():
192
  template_name = "llava_llama_2"
193
  elif "v1" in model_name.lower():
 
187
 
188
  if len(state.messages) == state.offset + 2:
189
  # First round of conversation
190
+ if "llava" or "mavila" in model_name.lower():
191
  if "llama-2" in model_name.lower():
192
  template_name = "llava_llama_2"
193
  elif "v1" in model_name.lower():