Skip to content

Commit df0e0f0

Browse files
authored
[CI/Build] Skip gpt_big model test because of broken HF model (#22848)
Signed-off-by: Isotr0py <[email protected]>
1 parent b4b78d6 commit df0e0f0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/models/registry.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,8 @@ def check_available_online(
195195
"GPT2LMHeadModel": _HfExamplesInfo("openai-community/gpt2",
196196
{"alias": "gpt2"}),
197197
"GPTBigCodeForCausalLM": _HfExamplesInfo("bigcode/starcoder",
198-
{"tiny": "bigcode/tiny_starcoder_py"}), # noqa: E501
198+
extras={"tiny": "bigcode/tiny_starcoder_py"}, # noqa: E501
199+
min_transformers_version="4.55.1"),
199200
"GPTJForCausalLM": _HfExamplesInfo("Milos/slovak-gpt-j-405M",
200201
{"6b": "EleutherAI/gpt-j-6b"}),
201202
"GPTNeoXForCausalLM": _HfExamplesInfo("EleutherAI/pythia-70m",

0 commit comments

Comments
 (0)