5b59532760
Co-authored-by: litianjian <litianjian@bytedance.com> Co-authored-by: Cyrus Leung <cyrus.tl.leung@gmail.com> Co-authored-by: Roger Wang <ywang@roblox.com> Co-authored-by: DarkLight1337 <tlleungac@connect.ust.hk>
16 lines
528 B
Python
16 lines
528 B
Python
import pytest
|
|
import transformers
|
|
|
|
from vllm.model_executor.models import _MODELS, ModelRegistry
|
|
|
|
|
|
@pytest.mark.parametrize("model_cls", _MODELS)
|
|
def test_registry_imports(model_cls):
|
|
if (model_cls in ("LlavaOnevisionForConditionalGeneration",
|
|
"Qwen2VLForConditionalGeneration")
|
|
and transformers.__version__ < "4.45"):
|
|
pytest.skip("Waiting for next transformers release")
|
|
|
|
# Ensure all model classes can be imported successfully
|
|
ModelRegistry.resolve_model_cls([model_cls])
|