From 76b4239ec867cbe2ec4030ff5cd14cb201b7ef87 Mon Sep 17 00:00:00 2001 From: KaranGarg <95518516+grgkaran03@users.noreply.github.com> Date: Fri, 14 Oct 2022 13:12:29 +0530 Subject: [PATCH] [Doctests] add `configuration_blenderbot_small.py` (#19589) * yoso config for doctest * Revert "yoso config for doctest" This reverts commit eae128d6f1b3631b676ffbcc181390e338819bd1. * add configurations_blenderbot_small.py for doctests --- .../models/blenderbot_small/configuration_blenderbot_small.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/blenderbot_small/configuration_blenderbot_small.py b/src/transformers/models/blenderbot_small/configuration_blenderbot_small.py index abeca877c6..0d24f9cc48 100644 --- a/src/transformers/models/blenderbot_small/configuration_blenderbot_small.py +++ b/src/transformers/models/blenderbot_small/configuration_blenderbot_small.py @@ -95,12 +95,12 @@ class BlenderbotSmallConfig(PretrainedConfig): Example: ```python - >>> from transformers import BlenderbotSmallModel, BlenderbotSmallConfig + >>> from transformers import BlenderbotSmallConfig, BlenderbotSmallModel >>> # Initializing a BlenderbotSmall facebook/blenderbot_small-90M style configuration >>> configuration = BlenderbotSmallConfig() - >>> # Initializing a model from the facebook/blenderbot_small-90M style configuration + >>> # Initializing a model (with random weights) from the facebook/blenderbot_small-90M style configuration >>> model = BlenderbotSmallModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index ec1093510e..6661dc4ede 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/configuration_blenderbot.py src/transformers/models/blenderbot/modeling_blenderbot.py +src/transformers/models/blenderbot_small/configuration_blenderbot_small.py src/transformers/models/blenderbot_small/modeling_blenderbot_small.py src/transformers/models/bloom/configuration_bloom.py src/transformers/models/canine/configuration_canine.py