Spaces:
Running
Running
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"
|