update

open-compass · Mar 23, 2024 · f92be3a · f92be3a
1 parent 7c9bcc5
commit f92be3a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vlmeval/vlm/llava.py b/vlmeval/vlm/llava.py
@@ -215,5 +215,7 @@ def generate(self, image_path, prompt, dataset=None):
 
         if '</s>' in answer:
             answer = answer.split('</s>')[0].strip()
+        if '<|im_end|>' in answer:
+            answer = answer.split('<|im_end|>')[0].strip()
 
         return answer