Unverified Commit 508baf19 authored by Thomas Chaigneau's avatar Thomas Chaigneau Committed by GitHub
Browse files

add bigbird typo fixes (#16897)


Co-authored-by: default avatarChainYo <t.chaigneau.tc@gmail.com>
parent 72728be3
......@@ -159,7 +159,7 @@ class FeaturesManager:
"question-answering",
onnx_config_cls=BertOnnxConfig,
),
"bigbird": supported_features_mapping(
"big-bird": supported_features_mapping(
"default",
"masked-lm",
"causal-lm",
......
......@@ -172,7 +172,7 @@ class OnnxConfigWithPastTestCaseV2(TestCase):
PYTORCH_EXPORT_MODELS = {
("albert", "hf-internal-testing/tiny-albert"),
("bert", "bert-base-cased"),
("bigbird", "google/bigbird-roberta-base"),
("big-bird", "google/bigbird-roberta-base"),
("ibert", "kssteven/ibert-roberta-base"),
("camembert", "camembert-base"),
("convbert", "YituTech/conv-bert-base"),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment