Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/onnx/features.py
View file @ 508baf19
...@@ -159,7 +159,7 @@ class FeaturesManager: ...@@ -159,7 +159,7 @@ class FeaturesManager:
"question-answering", "question-answering",
onnx_config_cls=BertOnnxConfig, onnx_config_cls=BertOnnxConfig,
), ),
"bigbird": supported_features_mapping( "big-bird": supported_features_mapping(
"default", "default",
"masked-lm", "masked-lm",
"causal-lm", "causal-lm",
......
tests/onnx/test_onnx_v2.py
View file @ 508baf19
...@@ -172,7 +172,7 @@ class OnnxConfigWithPastTestCaseV2(TestCase): ...@@ -172,7 +172,7 @@ class OnnxConfigWithPastTestCaseV2(TestCase):
PYTORCH_EXPORT_MODELS = { PYTORCH_EXPORT_MODELS = {
("albert", "hf-internal-testing/tiny-albert"), ("albert", "hf-internal-testing/tiny-albert"),
("bert", "bert-base-cased"), ("bert", "bert-base-cased"),
("bigbird", "google/bigbird-roberta-base"), ("big-bird", "google/bigbird-roberta-base"),
("ibert", "kssteven/ibert-roberta-base"), ("ibert", "kssteven/ibert-roberta-base"),
("camembert", "camembert-base"), ("camembert", "camembert-base"),
("convbert", "YituTech/conv-bert-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