update config for api models

shan23chen · Mar 19, 2024 · c2b66d3 · c2b66d3
1 parent fd4be1a
commit c2b66d3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/vlmeval/config.py b/vlmeval/config.py
@@ -49,22 +49,28 @@
  'InternVL-Chat-V1-1':partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-Chinese-V1-1'),
  'InternVL-Chat-V1-2': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-Chinese-V1-2'),
  'InternVL-Chat-V1-2-Plus': partial(InternVLChat, model_path='OpenGVLab/InternVL-Chat-Chinese-V1-2-Plus'),
- 'Claude3V':partial(Claude3V,model='claude-3-opus-20240229',temperature=0,retry=10),
 }
 
 api_models = {
  'GPT4V': partial(GPT4V, model='gpt-4-vision-preview', temperature=0, img_size=512, img_detail='low', retry=10),
+ # Internal Only
  'GPT4V_INT': partial(GPT4V_Internal, model='gpt-4-vision-preview', temperature=0, img_size=512, img_detail='low', retry=10),
  'GPT4V_SHORT': partial(
  GPT4V, model='gpt-4-vision-preview', temperature=0, img_size=512, img_detail='low', retry=10, 
  system_prompt="Please responde to the following question / request in a short reply. "),
+ # Internal Only
  'GPT4V_SHORT_INT': partial(
  GPT4V_Internal, model='gpt-4-vision-preview', temperature=0, img_size=512, img_detail='low', retry=10,
  system_prompt="Please responde to the following question / request in a short reply. "),
  'GeminiProVision': partial(GeminiProVision, temperature=0, retry=10),
  'QwenVLPlus': partial(QwenVLAPI, model='qwen-vl-plus', temperature=0, retry=10),
  'QwenVLMax': partial(QwenVLAPI, model='qwen-vl-max', temperature=0, retry=10),
+ # Internal Only
  'Step1V': partial(Step1V, temperature=0, retry=10),
+ # Internal Only
+ 'Claude3V_Opus': partial(Claude3V, model='claude-3-opus-20240229', temperature=0, retry=10),
+ 'Claude3V_Sonnet': partial(Claude3V, model='claude-3-sonnet-20240229', temperature=0, retry=10),
+ 'Claude3V_Haiku': partial(Claude3V, model='claude-3-haiku-20240307', temperature=0, retry=10),
 }
 
 xtuner_models = {