Update libra/eval/run_libra.py
Browse files- libra/eval/run_libra.py +4 -2
libra/eval/run_libra.py
CHANGED
|
@@ -159,9 +159,11 @@ def libra_eval(
|
|
| 159 |
else:
|
| 160 |
qs = DEFAULT_IMAGE_TOKEN + '\n' + qs
|
| 161 |
|
| 162 |
-
if '
|
| 163 |
mode_conv = "libra_llama_3"
|
| 164 |
-
|
|
|
|
|
|
|
| 165 |
mode_conv = "mistral_instruct"
|
| 166 |
else:
|
| 167 |
mode_conv = "libra_v1"
|
|
|
|
| 159 |
else:
|
| 160 |
qs = DEFAULT_IMAGE_TOKEN + '\n' + qs
|
| 161 |
|
| 162 |
+
if 'libra-v1.0-3b' in model_name.lower():
|
| 163 |
mode_conv = "libra_llama_3"
|
| 164 |
+
elif 'cxrgen' in model_name.lower():
|
| 165 |
+
mode_conv = "libra_v0"
|
| 166 |
+
elif 'mistral' in model_name.lower():
|
| 167 |
mode_conv = "mistral_instruct"
|
| 168 |
else:
|
| 169 |
mode_conv = "libra_v1"
|