[Bugfix][Model] Add base class for vision-language models (#4809)
This commit is contained in:
9
tests/models/test_registry.py
Normal file
9
tests/models/test_registry.py
Normal 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)
|
||||
Reference in New Issue
Block a user