fix BlenderbotSmallTokenizer (#9538)

* add model_input_names * fix test
2021-01-13 10:53:43 +05:30
parent 2df34f4aba
commit 69ed36063a
2 changed files with 1 additions and 4 deletions
--- a/src/transformers/models/blenderbot_small/tokenization_blenderbot_small.py
+++ b/src/transformers/models/blenderbot_small/tokenization_blenderbot_small.py
@@ -92,6 +92,7 @@ class BlenderbotSmallTokenizer(PreTrainedTokenizer):
        },
    }
    max_model_input_sizes = {"facebook/blenderbot_small-90M": 512}
+    model_input_names = ["attention_mask"]

    def __init__(
        self,