Update libra/eval/run_libra.py

libra/eval/run_libra.py

@@ -159,7 +159,11 @@ def libra_eval(
     else:
         qs = DEFAULT_IMAGE_TOKEN + '\n' + qs
 
-    if 'libra' in model_name.lower():
+    if 'llama-3' in model_name.lower():
+        mode_conv = "libra_llama_3"
+    if 'mistral' in model_name.lower():
+        mode_conv = "mistral_instruct"
+    else:
         mode_conv = "libra_v1"
 
     if conv_mode is not None and mode_conv != conv_mode:
@@ -178,7 +182,7 @@ def libra_eval(
 
     image_tensor = get_image_tensors(image_file, image_processor, model, device=device)
     
-    stop_str = conv.sep if conv.sep_style != SeparatorStyle.TWO else conv.sep2
+    stop_str = conv.sep if conv.sep_style not in {SeparatorStyle.TWO, SeparatorStyle.LLAMA_3} else conv.sep2
     keywords = [stop_str]
     stopping_criteria = KeywordsStoppingCriteria(keywords, tokenizer, input_ids)