add bigbird typo fixes (#16897)
Co-authored-by: ChainYo <t.chaigneau.tc@gmail.com>
This commit is contained in:
@@ -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"),
|
||||
|
||||
Reference in New Issue
Block a user