From 4a5d63c9581ca93fe36f68f49993814127a7d702 Mon Sep 17 00:00:00 2001 From: imarekkus <49692939+imarekkus@users.noreply.github.com> Date: Wed, 12 Oct 2022 20:02:55 +0200 Subject: [PATCH] Albert config update (#19541) --- src/transformers/models/albert/configuration_albert.py | 2 +- utils/documentation_tests.txt | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/models/albert/configuration_albert.py b/src/transformers/models/albert/configuration_albert.py index 1f2c5ca3c8..d11d49a2c7 100644 --- a/src/transformers/models/albert/configuration_albert.py +++ b/src/transformers/models/albert/configuration_albert.py @@ -101,7 +101,7 @@ class AlbertConfig(PretrainedConfig): ... intermediate_size=3072, ... ) - >>> # Initializing a model from the ALBERT-base style configuration + >>> # Initializing a model (with random weights) from the ALBERT-base style configuration >>> model = AlbertModel(albert_xxlarge_configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index e87b8746c7..d2465276b0 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -12,6 +12,7 @@ docs/source/en/model_doc/tapex.mdx docs/source/en/model_doc/donut.mdx docs/source/en/model_doc/encoder-decoder.mdx src/transformers/generation_utils.py +src/transformers/models/albert/configuration_albert.py src/transformers/models/albert/modeling_albert.py src/transformers/models/albert/modeling_tf_albert.py src/transformers/models/bart/configuration_bart.py