OpenCompass/opencompass/multimodal/models/llava/__init__.py
Yike Yuan 3f601f420b
[Feat] Support public dataset of visualglm and llava. (#265)
* [Feat] Add public dataset support of VisualGLM.

* [Feat] Refactor LLaVA.

* [Feat] Add public dataset support of LlaVA.

* [Fix] Add  arg.
2023-08-25 15:44:32 +08:00

13 lines
563 B
Python

from .llava import LLaVA
from .post_processor import LLaVABasePostProcessor, LLaVAVSRPostProcessor
from .prompt_constructor import (LLaVABasePromptConstructor,
LLaVAMMBenchPromptConstructor,
LLaVAScienceQAPromptConstructor,
LLaVAVQAPromptConstructor)
__all__ = [
'LLaVA', 'LLaVABasePromptConstructor', 'LLaVAMMBenchPromptConstructor',
'LLaVABasePostProcessor', 'LLaVAVQAPromptConstructor',
'LLaVAScienceQAPromptConstructor', 'LLaVAVSRPostProcessor'
]