From 508baf194313c397345af868202404e285494a28 Mon Sep 17 00:00:00 2001 From: Thomas Chaigneau <50595514+ChainYo@users.noreply.github.com> Date: Mon, 25 Apr 2022 11:32:06 +0200 Subject: [PATCH] add bigbird typo fixes (#16897) Co-authored-by: ChainYo --- src/transformers/onnx/features.py | 2 +- tests/onnx/test_onnx_v2.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/onnx/features.py b/src/transformers/onnx/features.py index 9163de8c21..31bf0b45d7 100644 --- a/src/transformers/onnx/features.py +++ b/src/transformers/onnx/features.py @@ -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", diff --git a/tests/onnx/test_onnx_v2.py b/tests/onnx/test_onnx_v2.py index f85fe71218..e8b403e54f 100644 --- a/tests/onnx/test_onnx_v2.py +++ b/tests/onnx/test_onnx_v2.py @@ -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"),