mirror of
https://github.com/OpenBMB/MiniCPM-V.git
synced 2026-02-04 09:49:20 +08:00
21 lines
621 B
Python
21 lines
621 B
Python
from vlmeval.vlm import *
|
|
from vlmeval.api import *
|
|
from functools import partial
|
|
|
|
minicpm_series = {
|
|
'MiniCPM-V': partial(MiniCPM_V, model_path='openbmb/MiniCPM-V'),
|
|
'MiniCPM-V-2': partial(MiniCPM_V, model_path='openbmb/MiniCPM-V-2'),
|
|
'MiniCPM-Llama3-V-2_5': partial(MiniCPM_Llama3_V, model_path='openbmb/MiniCPM-Llama3-V-2_5'),
|
|
'MiniCPM-V-2_6': partial(MiniCPM_V_2_6, model_path='openbmb/MiniCPM-V-2_6'),
|
|
'MiniCPM-o-2_6': partial(MiniCPM_o_2_6, model_path='openbmb/MiniCPM-o-2_6'),
|
|
}
|
|
|
|
supported_VLM = {}
|
|
|
|
model_groups = [
|
|
minicpm_series
|
|
]
|
|
|
|
for grp in model_groups:
|
|
supported_VLM.update(grp)
|