Tokenization + small fixes

2019-10-30 14:39:16 +00:00
parent 67b422662c
commit fedac786d4
2 changed files with 211 additions and 1 deletions
--- a/transformers/modeling_albert.py
+++ b/transformers/modeling_albert.py
@@ -298,7 +298,7 @@ ALBERT_INPUTS_DOCSTRING = r"""
 """

@add_start_docstrings("The bare ALBERT Model transformer outputting raw hidden-states without any specific head on top.",
-                      BERT_START_DOCSTRING, BERT_INPUTS_DOCSTRING)
+                      ALBERT_START_DOCSTRING, ALBERT_INPUTS_DOCSTRING)
 class AlbertModel(BertModel):
    def __init__(self, config):
        super(AlbertModel, self).__init__(config)