X-iZhang commited on
Commit
bf242cc
·
verified ·
1 Parent(s): b3f37a7

Update libra/eval/run_libra.py

Browse files
Files changed (1) hide show
  1. 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 'llama-3' in model_name.lower():
163
  mode_conv = "libra_llama_3"
164
- if 'mistral' in model_name.lower():
 
 
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"