2024-05-19 15:13:33 +08:00
|
|
|
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
|
2024-08-06 16:55:31 +08:00
|
|
|
ModelRegistry.resolve_model_cls([model_cls])
|