Skip to content

Commit 8276984

Browse files
committed
Fix wrong template
1 parent 07ef6e9 commit 8276984

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

internvl_chat_llava/llava/serve/gradio_web_server.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -201,11 +201,11 @@ def http_bot(state, model_selector, temperature, top_p, max_new_tokens, max_inpu
201201
else:
202202
template_name = "llava_v0"
203203
elif "intern" in model_name.lower():
204-
if "hermes2" in model_name.lower() or "v1-2" in model_name.lower():
204+
if any(x in model_name.lower() for x in ["hermes2", "v1-2", "v1_2"]):
205205
template_name = "Hermes-2"
206-
elif "internlm2" in model_name.lower() or "v1-5" in model_name.lower():
206+
elif any(x in model_name.lower() for x in ["internlm2", "v1-5", "v1_5"]):
207207
template_name = "internlm2-chat"
208-
elif "chinese" in model_name.lower() or "v1-1" in model_name.lower():
208+
elif any(x in model_name.lower() for x in ["chinese", "v1-1", "v1_1"]):
209209
template_name = "internvl_zh"
210210
else:
211211
template_name = "llava_v1"

0 commit comments

Comments
 (0)