mirror of
https://github.com/wassname/vllm.git
synced 2026-09-09 11:40:01 +08:00
[Bugfix][Model] Add base class for vision-language models (#4809)
This commit is contained in:
@@ -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