Update llava/serve/gradio_web_server.py

llava/serve/gradio_web_server.py

@@ -164,7 +164,7 @@ def http_bot(state, model_selector, temperature, top_p, max_new_tokens, request:
 
     if len(state.messages) == state.offset + 2:
         # First round of conversation
-        if "llava" in model_name.lower():
+        if "llava" or "mavila" in model_name.lower():
             if 'llama-2' in model_name.lower():
                 template_name = "llava_llama_2"
             elif "v1" in model_name.lower():