We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b4b78d6 commit df0e0f0Copy full SHA for df0e0f0
tests/models/registry.py
@@ -195,7 +195,8 @@ def check_available_online(
195
"GPT2LMHeadModel": _HfExamplesInfo("openai-community/gpt2",
196
{"alias": "gpt2"}),
197
"GPTBigCodeForCausalLM": _HfExamplesInfo("bigcode/starcoder",
198
- {"tiny": "bigcode/tiny_starcoder_py"}), # noqa: E501
+ extras={"tiny": "bigcode/tiny_starcoder_py"}, # noqa: E501
199
+ min_transformers_version="4.55.1"),
200
"GPTJForCausalLM": _HfExamplesInfo("Milos/slovak-gpt-j-405M",
201
{"6b": "EleutherAI/gpt-j-6b"}),
202
"GPTNeoXForCausalLM": _HfExamplesInfo("EleutherAI/pythia-70m",
0 commit comments