"tests/models/multimodal/processing/test_idefics3.py" did not exist on "eed11ebee93e9d137ac74d8e6e97427354bd3797"
test_registry.py 462 Bytes
Newer Older
1
import pytest
2
import transformers
3
4
5
6
7
8

from vllm.model_executor.models import _MODELS, ModelRegistry


@pytest.mark.parametrize("model_cls", _MODELS)
def test_registry_imports(model_cls):
9
10
11
12
    if (model_cls == "Qwen2VLForConditionalGeneration"
            and transformers.__version__ < "4.45"):
        pytest.skip("Waiting for next transformers release")

13
    # Ensure all model classes can be imported successfully
14
    ModelRegistry.resolve_model_cls([model_cls])