diff --git a/tests/models/registry.py b/tests/models/registry.py index d7d20d1f3..eb48c0f6a 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",