[Doctest] GPTNeoConfig , GPTNeoXConfig , GPTNeoXJapaneseConfig (huggi…

…ngface#19741)
impira · Oct 19, 2022 · 7df0751 · 7df0751
1 parent 71786b1
commit 7df0751
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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