[CI] Skip the pooling models that do not support transformers v4.55 (#22411)

Signed-off-by: wang.yuqi <noooop@126.com>
2025-08-07 14:05:03 +08:00
parent 4be02a3776
commit 2a4c825523
4 changed files with 28 additions and 1 deletions
--- a/tests/models/utils.py
+++ b/tests/models/utils.py
@@ -412,3 +412,14 @@ def dummy_hf_overrides(
        })

    return hf_config
+
+
+def check_transformers_version(model: str,
+                               min_transformers_version: Optional[str] = None,
+                               max_transformers_version: Optional[str] = None):
+    from .registry import _HfExamplesInfo
+
+    return _HfExamplesInfo(model,
+                           min_transformers_version=min_transformers_version,
+                           max_transformers_version=max_transformers_version
+                           ).check_transformers_version(on_fail="skip")