Implementation:Open compass VLMEvalKit Yi VL
Appearance
| Field | Value |
|---|---|
| source | VLMEvalKit |
| domain | Vision, Model_Architecture |
Overview
VLM adapter for the Yi-VL model enabling benchmark evaluation in VLMEvalKit.
Description
Yi_VL inherits from BaseModel and wraps the Yi-VL model for use within the VLMEvalKit evaluation framework. It initializes the model and tokenizer/processor from a HuggingFace model path (default: 01-ai/Yi-VL-6B) and provides the generate_inner method for inference. Requires the Yi repository to be cloned and configured with Yi_ROOT path.
Usage
Register in vlmeval/config.py via supported_VLM and invoke through the standard evaluation pipeline.
Code Reference
- Source:
vlmeval/vlm/yi_vl.py, Lines: L1-156 - Import:
from vlmeval.vlm.yi_vl import Yi_VL
Signature:
class Yi_VL(BaseModel):
INSTALL_REQ = True
INTERLEAVE = False
def __init__(self, model_path='01-ai/Yi-VL-6B', **kwargs): ...
def generate_inner(self, message, dataset=None): ...
I/O Contract
| Direction | Description |
|---|---|
| Inputs | message — list of dicts with type (text/image) and value; dataset — optional dataset name for custom prompting |
| Outputs | generate_inner() returns str (model response text) |
Usage Examples
from vlmeval.vlm.yi_vl import Yi_VL
model = Yi_VL(model_path='path/to/model')
response = model.generate_inner(message)
Related Pages
Page Connections
Double-click a node to navigate. Hold to expand connections.
Principle
Implementation
Heuristic
Environment