From df0e0f023e1be63d259280ccb9caf5547302ad30 Mon Sep 17 00:00:00 2001 From: Isotr0py Date: Thu, 14 Aug 2025 04:36:28 +0800 Subject: [PATCH] [CI/Build] Skip gpt_big model test because of broken HF model (#22848) Signed-off-by: Isotr0py --- tests/models/registry.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/models/registry.py b/tests/models/registry.py index d7d20d1f3abf7..eb48c0f6a7738 100644 --- a/tests/models/registry.py +++ b/tests/models/registry.py @@ -195,7 +195,8 @@ _TEXT_GENERATION_EXAMPLE_MODELS = { "GPT2LMHeadModel": _HfExamplesInfo("openai-community/gpt2", {"alias": "gpt2"}), "GPTBigCodeForCausalLM": _HfExamplesInfo("bigcode/starcoder", - {"tiny": "bigcode/tiny_starcoder_py"}), # noqa: E501 + extras={"tiny": "bigcode/tiny_starcoder_py"}, # noqa: E501 + min_transformers_version="4.55.1"), "GPTJForCausalLM": _HfExamplesInfo("Milos/slovak-gpt-j-405M", {"6b": "EleutherAI/gpt-j-6b"}), "GPTNeoXForCausalLM": _HfExamplesInfo("EleutherAI/pythia-70m",