Spaces:
Sleeping
Sleeping
Update llava/serve/gradio_web_server.py
Browse files
llava/serve/gradio_web_server.py
CHANGED
@@ -164,7 +164,7 @@ def http_bot(state, model_selector, temperature, top_p, max_new_tokens, request:
|
|
164 |
|
165 |
if len(state.messages) == state.offset + 2:
|
166 |
# First round of conversation
|
167 |
-
if "llava" in model_name.lower():
|
168 |
if 'llama-2' in model_name.lower():
|
169 |
template_name = "llava_llama_2"
|
170 |
elif "v1" in model_name.lower():
|
|
|
164 |
|
165 |
if len(state.messages) == state.offset + 2:
|
166 |
# First round of conversation
|
167 |
+
if "llava" or "mavila" in model_name.lower():
|
168 |
if 'llama-2' in model_name.lower():
|
169 |
template_name = "llava_llama_2"
|
170 |
elif "v1" in model_name.lower():
|