From 7df0751cc623cc5a16321c344183718f0d9e82f8 Mon Sep 17 00:00:00 2001 From: Rohith ND <73429989+ndrohith09@users.noreply.github.com> Date: Wed, 19 Oct 2022 17:52:41 +0530 Subject: [PATCH] [Doctest] GPTNeoConfig , GPTNeoXConfig , GPTNeoXJapaneseConfig (#19741) --- src/transformers/models/gpt_neo/configuration_gpt_neo.py | 4 ++-- src/transformers/models/gpt_neox/configuration_gpt_neox.py | 4 ++-- .../gpt_neox_japanese/configuration_gpt_neox_japanese.py | 4 ++-- utils/documentation_tests.txt | 3 +++ 4 files changed, 9 insertions(+), 6 deletions(-) diff --git a/src/transformers/models/gpt_neo/configuration_gpt_neo.py b/src/transformers/models/gpt_neo/configuration_gpt_neo.py index 00054a2c6b..4d07670621 100644 --- a/src/transformers/models/gpt_neo/configuration_gpt_neo.py +++ b/src/transformers/models/gpt_neo/configuration_gpt_neo.py @@ -82,12 +82,12 @@ class GPTNeoConfig(PretrainedConfig): Example: ```python - >>> from transformers import GPTNeoModel, GPTNeoConfig + >>> from transformers import GPTNeoConfig, GPTNeoModel >>> # Initializing a GPTNeo EleutherAI/gpt-neo-1.3B style configuration >>> configuration = GPTNeoConfig() - >>> # Initializing a model from the EleutherAI/gpt-neo-1.3B style configuration + >>> # Initializing a model (with random weights) from the EleutherAI/gpt-neo-1.3B style configuration >>> model = GPTNeoModel(configuration) >>> # Accessing the model configuration diff --git a/src/transformers/models/gpt_neox/configuration_gpt_neox.py b/src/transformers/models/gpt_neox/configuration_gpt_neox.py index 3ce1bd7873..966c396189 100644 --- a/src/transformers/models/gpt_neox/configuration_gpt_neox.py +++ b/src/transformers/models/gpt_neox/configuration_gpt_neox.py @@ -72,12 +72,12 @@ class GPTNeoXConfig(PretrainedConfig): Example: ```python - >>> from transformers import GPTNeoXModel, GPTNeoXConfig + >>> from transformers import GPTNeoXConfig, GPTNeoXModel >>> # Initializing a GPTNeoX gpt-neox-20b style configuration >>> configuration = GPTNeoXConfig() - >>> # Initializing a model from the gpt-neox-20b style configuration + >>> # Initializing a model (with random weights) from the gpt-neox-20b style configuration >>> model = GPTNeoXModel(configuration) >>> # Accessing the model configuration diff --git a/src/transformers/models/gpt_neox_japanese/configuration_gpt_neox_japanese.py b/src/transformers/models/gpt_neox_japanese/configuration_gpt_neox_japanese.py index 749a9400b9..1cf08ef301 100644 --- a/src/transformers/models/gpt_neox_japanese/configuration_gpt_neox_japanese.py +++ b/src/transformers/models/gpt_neox_japanese/configuration_gpt_neox_japanese.py @@ -72,12 +72,12 @@ class GPTNeoXJapaneseConfig(PretrainedConfig): Example: ```python - >>> from transformers import GPTNeoXJapaneseModel, GPTNeoXJapaneseConfig + >>> from transformers import GPTNeoXJapaneseConfig, GPTNeoXJapaneseModel >>> # Initializing a GPTNeoXJapanese gpt-neox-japanese-2.7b style configuration >>> configuration = GPTNeoXJapaneseConfig() - >>> # Initializing a model from the gpt-neox-japanese-2.7b style configuration + >>> # Initializing a model (with random weights) from the gpt-neox-japanese-2.7b style configuration >>> model = GPTNeoXJapaneseModel(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 9782c80015..a6a4f789de 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -67,6 +67,9 @@ src/transformers/models/glpn/modeling_glpn.py src/transformers/models/gpt2/configuration_gpt2.py src/transformers/models/gpt2/modeling_gpt2.py src/transformers/models/gptj/modeling_gptj.py +src/transformers/models/gpt_neo/configuration_gpt_neo.py +src/transformers/models/gpt_neox/configuration_gpt_neox.py +src/transformers/models/gpt_neox_japanese/configuration_gpt_neox_japanese.py src/transformers/models/groupvit/modeling_groupvit.py src/transformers/models/groupvit/modeling_tf_groupvit.py src/transformers/models/hubert/modeling_hubert.py