SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

add bigbird typo fixes (#16897)

Browse Source 
Co-authored-by: ChainYo <t.chaigneau.tc@gmail.com>
This commit is contained in:
Thomas Chaigneau
2022-04-25 11:32:06 +02:00
committed by GitHub
parent 72728be3db
commit 508baf1943
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/onnx/features.py
View File

@@ -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",

2
tests/onnx/test_onnx_v2.py
View File

@@ -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"),