diff --git a/src/transformers/models/levit/configuration_levit.py b/src/transformers/models/levit/configuration_levit.py index b0258a56c2d..525221217ad 100644 --- a/src/transformers/models/levit/configuration_levit.py +++ b/src/transformers/models/levit/configuration_levit.py @@ -76,12 +76,12 @@ class LevitConfig(PretrainedConfig): Example: ```python - >>> from transformers import LevitModel, LevitConfig + >>> from transformers import LevitConfig, LevitModel >>> # Initializing a LeViT levit-128S style configuration >>> configuration = LevitConfig() - >>> # Initializing a model from the levit-128S style configuration + >>> # Initializing a model (with random weights) from the levit-128S style configuration >>> model = LevitModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index db609b8b7bf..7935a743c47 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -64,6 +64,7 @@ src/transformers/models/layoutlm/modeling_tf_layoutlm.py src/transformers/models/layoutlmv2/modeling_layoutlmv2.py src/transformers/models/layoutlmv3/modeling_layoutlmv3.py src/transformers/models/layoutlmv3/modeling_tf_layoutlmv3.py +src/transformers/models/levit/configuration_levit.py src/transformers/models/lilt/modeling_lilt.py src/transformers/models/longformer/modeling_longformer.py src/transformers/models/longformer/modeling_tf_longformer.py