aphrodite-engine/tests/models/test_registry.py

import pytest
import transformers

from aphrodite.modeling.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])