From 05e9b97663982d1555d9f4205a6eaaa936ffb12c Mon Sep 17 00:00:00 2001 From: kennymckormick Date: Sat, 23 Mar 2024 14:02:41 +0800 Subject: [PATCH] update --- vlmeval/vlm/llava.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vlmeval/vlm/llava.py b/vlmeval/vlm/llava.py index 4c03a8f53..cd9083183 100644 --- a/vlmeval/vlm/llava.py +++ b/vlmeval/vlm/llava.py @@ -211,7 +211,7 @@ def generate(self, image_path, prompt, dataset=None): elif 'ASSISTANT:' in answer: answer = answer.split('ASSISTANT:')[1].strip() elif '<|im_start|>assistant\n' in answer: - answer = answer.split('<|im_start|>assistant\n')[1].strip() + answer = answer.split('assistant\n')[1].strip() if '' in answer: answer = answer.split('')[0].strip()