Implementation:Open compass VLMEvalKit Cosmos

Field	Value
source	VLMEvalKit
domain	Vision, Model_Architecture

Overview

VLM adapter for the Cosmos-Reason model enabling benchmark evaluation in VLMEvalKit.

Description

Cosmos inherits from BaseModel and wraps the Cosmos-Reason model for use within the VLMEvalKit evaluation framework. It initializes the model and tokenizer/processor from a HuggingFace model path (default: nvidia/Cosmos-Reason1-7B) and provides the generate_inner method for inference. Uses vLLM for inference with configurable sampling parameters.

Usage

Register in vlmeval/config.py via supported_VLM and invoke through the standard evaluation pipeline.

Code Reference

Source: vlmeval/vlm/cosmos.py, Lines: L1-77
Import: from vlmeval.vlm.cosmos import Cosmos

Signature:

class Cosmos(BaseModel):
    INSTALL_REQ = False
    INTERLEAVE = True
    def __init__(self, model_path='nvidia/Cosmos-Reason1-7B', **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.cosmos import Cosmos
model = Cosmos(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