[Bugfix][Model] Add base class for vision-language models (#4809)

This commit is contained in:
Cyrus Leung
2024-05-19 15:13:33 +08:00
committed by GitHub
parent 2e9a2227ec
commit f68470e803
4 changed files with 53 additions and 29 deletions

View File

@ -0,0 +1,9 @@
import pytest
from vllm.model_executor.models import _MODELS, ModelRegistry
@pytest.mark.parametrize("model_cls", _MODELS)
def test_registry_imports(model_cls):
# Ensure all model classes can be imported successfully
ModelRegistry.load_model_cls(model_cls)