From 094a50dfd8dd55c80acca12b7ce3ed43f38a1271 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Fri, 6 Oct 2023 17:09:56 +0200 Subject: [PATCH] example fix docstring --- src/transformers/models/albert/configuration_albert.py | 6 ++++++ utils/check_docstrings.py | 1 - 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/transformers/models/albert/configuration_albert.py b/src/transformers/models/albert/configuration_albert.py index fd0c62388792..cacc0499035c 100644 --- a/src/transformers/models/albert/configuration_albert.py +++ b/src/transformers/models/albert/configuration_albert.py @@ -85,6 +85,12 @@ class AlbertConfig(PretrainedConfig): [Self-Attention with Relative Position Representations (Shaw et al.)](https://arxiv.org/abs/1803.02155). For more information on `"relative_key_query"`, please refer to *Method 4* in [Improve Transformer Models with Better Relative Position Embeddings (Huang et al.)](https://arxiv.org/abs/2009.13658). + pad_token_id (`int`, *optional*, defaults to 0): + Padding token id. + bos_token_id (`int`, *optional*, defaults to 2): + Beginning of stream token id. + eos_token_id (`int`, *optional*, defaults to 3): + End of stream token id. Examples: diff --git a/utils/check_docstrings.py b/utils/check_docstrings.py index 8d69ea603f15..650b519eaa57 100644 --- a/utils/check_docstrings.py +++ b/utils/check_docstrings.py @@ -75,7 +75,6 @@ "TFGPT2Tokenizer", # Missing arguments in the docstring "ASTFeatureExtractor", - "AlbertConfig", "AlbertModel", "AlbertTokenizerFast", "AlignTextModel",