update

open-compass · Mar 23, 2024 · 05e9b97 · 05e9b97
1 parent 837a2a8
commit 05e9b97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 '</s>' in answer:
             answer = answer.split('</s>')[0].strip()