Function check_transformers_version

scripts/generate_tiny_models/_common.py:58–71

Raise unless the installed transformers matches `expected_version` exactly. If `expected_version` is None, defaults to the `transformers>=` floor from pyproject.toml.

(expected_version=None)

def check_transformers_version(expected_version=None):
    """Raise unless the installed transformers matches `expected_version` exactly.

    If `expected_version` is None, defaults to the `transformers>=` floor from pyproject.toml.
    """
    import transformers

    if expected_version is None:
        expected_version = _transformers_floor_from_pyproject()

    if Version(transformers.__version__) != Version(expected_version):
        raise RuntimeError(
            f"This script requires transformers=={expected_version}, but {transformers.__version__} is installed."
        )


def smoke_test(model, tokenizer_or_processor=None):