mirror of
https://github.com/OpenBMB/MiniCPM-V.git
synced 2026-02-05 02:09:20 +08:00
20 lines
447 B
Python
20 lines
447 B
Python
from vlmeval.vlm import *
|
|
from vlmeval.api import *
|
|
from functools import partial
|
|
|
|
ungrouped = {
|
|
'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'),
|
|
}
|
|
|
|
supported_VLM = {}
|
|
|
|
model_groups = [
|
|
ungrouped
|
|
]
|
|
|
|
for grp in model_groups:
|
|
supported_VLM.update(grp)
|
|
|