Fix run_llava.py

NVlabs · Mar 13, 2024 · 9ae25d4 · 9ae25d4
1 parent 3d6f1a6
commit 9ae25d4
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/llava/eval/run_llava.py b/llava/eval/run_llava.py
@@ -67,10 +67,11 @@ def eval_model(args):
         else:
             qs = re.sub(IMAGE_PLACEHOLDER, DEFAULT_IMAGE_TOKEN, qs)
     else:
-        if model.config.mm_use_im_start_end:
-            qs = image_token_se + "\n" + qs
-        else:
-            qs = DEFAULT_IMAGE_TOKEN + "\n" + qs
+        if DEFAULT_IMAGE_TOKEN not in qs:
+            if model.config.mm_use_im_start_end:
+                qs = image_token_se + "\n" + qs
+            else:
+                qs = DEFAULT_IMAGE_TOKEN + "\n" + qs
 
     if "llama-2" in model_name.lower():
         conv_mode = "llava_llama_2"