[CI Failure] Add transformers version check for openai/privacy-filter (#47011)
Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>
wang.yuqi committed
Jun 29, 2026 at 16:57 UTC
9e86352c606c61095029f156ae3e4ac2097cf7e5
1 file changed
+4
tests/models/language/pooling/test_token_classification.py
+4
@@ -5,6 +5,7 @@ import pytest
5
import torch
6
from transformers import AutoModelForTokenClassification
7
8
+from tests.models.registry import HF_EXAMPLE_MODELS
9
from tests.models.utils import softmax
10
from vllm.platforms import current_platform
11
from vllm.utils.torch_utils import set_random_seed
@@ -136,6 +137,9 @@ def test_openai_privacy_filter(
137
model: str,
138
dtype: str,
139
) -> None:
140
+ model_info = HF_EXAMPLE_MODELS.find_hf_info(model)
141
+ model_info.check_transformers_version(on_fail="skip")
142
+
143
with vllm_runner(model, max_model_len=None, dtype=dtype) as vllm_model:
144
vllm_outputs = vllm_model.token_classify(PRIVACY_FILTER_PROMPTS)
145