From 0903fc80b5fd17ff5b9350fc497a3ed381800656 Mon Sep 17 00:00:00 2001 From: imarekkus <49692939+imarekkus@users.noreply.github.com> Date: Thu, 13 Oct 2022 12:14:38 +0200 Subject: [PATCH] [Doctest] bloom config update (#19566) --- src/transformers/models/bloom/configuration_bloom.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/bloom/configuration_bloom.py b/src/transformers/models/bloom/configuration_bloom.py index 4f973d93ae..9b77d17fd2 100644 --- a/src/transformers/models/bloom/configuration_bloom.py +++ b/src/transformers/models/bloom/configuration_bloom.py @@ -93,12 +93,12 @@ class BloomConfig(PretrainedConfig): Example: ```python - >>> from transformers import BloomModel, BloomConfig + >>> from transformers import BloomConfig, BloomModel >>> # Initializing a Bloom configuration >>> configuration = BloomConfig() - >>> # Initializing a model from the configuration + >>> # Initializing a model (with random weights) from the configuration >>> model = BloomModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 9ad5ca2ba7..4c15f84fb0 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -27,6 +27,7 @@ src/transformers/models/bigbird_pegasus/modeling_bigbird_pegasus.py src/transformers/models/big_bird/modeling_big_bird.py src/transformers/models/blenderbot/modeling_blenderbot.py src/transformers/models/blenderbot_small/modeling_blenderbot_small.py +src/transformers/models/bloom/configuration_bloom.py src/transformers/models/conditional_detr/modeling_conditional_detr.py src/transformers/models/convnext/modeling_convnext.py src/transformers/models/ctrl/modeling_ctrl.py