mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
191a3f6f9d
* [Feature]: Add minigpt-4 * [Feature]: Add mm local runner * [Feature]: Add instructblip * [Feature]: Delete redundant file * [Feature]: Delete redundant file * [Feature]: Add README to InstructBLIP * [Feature]: Update MiniGPT-4 * [Fix]: Fix lint * [Feature]add omnibenchmark readme (#49) * add omnibenchmark readme * fix * Update OmniMMBench.md * Update OmniMMBench.md * Update OmniMMBench.md * [Fix]: Refine name (#54) * [Feature]: Unify out and err * [Fix]: Fix lint * [Feature]: Rename to mmbench and change weight path * [Feature]: Delete Omni in instructblip * [Feature]: Check the avaliablity of lavis * [Fix]: Fix lint * [Feature]: Refactor MM * [Refactor]: Refactor path * [Feature]: Delete redundant files * [Refactor]: Delete redundant files --------- Co-authored-by: Wangbo Zhao(黑色枷锁) <56866854+wangbo-zhao@users.noreply.github.com>
37 lines
1.7 KiB
Python
37 lines
1.7 KiB
Python
from mmengine.registry import DATASETS as MMENGINE_DATASETS
|
|
from mmengine.registry import METRICS as MMENGINE_METRICS
|
|
from mmengine.registry import MODELS as MMENGINE_MODELS
|
|
from mmengine.registry import Registry
|
|
|
|
PARTITIONERS = Registry('partitioner', locations=['opencompass.partitioners'])
|
|
RUNNERS = Registry('runner', locations=['opencompass.runners'])
|
|
TASKS = Registry('task', locations=['opencompass.tasks'])
|
|
MODELS = Registry('model', locations=['opencompass.models'])
|
|
# TODO: LOAD_DATASET -> DATASETS
|
|
LOAD_DATASET = Registry('load_dataset', locations=['opencompass.datasets'])
|
|
TEXT_POSTPROCESSORS = Registry(
|
|
'text_postprocessors', locations=['opencompass.utils.text_postprocessors'])
|
|
EVALUATORS = Registry('evaluators', locations=['opencompass.evaluators'])
|
|
|
|
ICL_INFERENCERS = Registry('icl_inferencers',
|
|
locations=['opencompass.openicl.icl_inferencer'])
|
|
ICL_RETRIEVERS = Registry('icl_retrievers',
|
|
locations=['opencompass.openicl.icl_retriever'])
|
|
ICL_DATASET_READERS = Registry(
|
|
'icl_dataset_readers',
|
|
locations=['opencompass.openicl.icl_dataset_reader'])
|
|
ICL_PROMPT_TEMPLATES = Registry(
|
|
'icl_prompt_templates',
|
|
locations=['opencompass.openicl.icl_prompt_template'])
|
|
ICL_EVALUATORS = Registry('icl_evaluators',
|
|
locations=['opencompass.openicl.icl_evaluator'])
|
|
DATASETS = Registry('mm_datasets',
|
|
parent=MMENGINE_DATASETS,
|
|
locations=['opencompass.multimodal.datasets'])
|
|
METRICS = Registry('metric',
|
|
parent=MMENGINE_METRICS,
|
|
locations=['opencompass.metrics'])
|
|
MM_MODELS = Registry('mm_model',
|
|
parent=MMENGINE_MODELS,
|
|
locations=['opencompass.multimodal.models'])
|