[GPTJ] fix docs (#15558)

0acd84f7 · Suraj Patil · GitHub · 87d08afb · 0acd84f7 · 0acd84f7
Unverified Commit 0acd84f7 authored Feb 08, 2022 by Suraj Patil Committed by GitHub Feb 08, 2022
Showing with 2 additions and 4 deletions

src/transformers/models/gptj/modeling_flax_gptj.py src/transformers/models/gptj/modeling_flax_gptj.py +1 -2

src/transformers/models/gptj/modeling_gptj.py src/transformers/models/gptj/modeling_gptj.py +1 -2

No files found.
--- a/src/transformers/models/gptj/modeling_flax_gptj.py
+++ b/src/transformers/models/gptj/modeling_flax_gptj.py
@@ -668,8 +668,7 @@ class FlaxGPTJForCausalLMModule(nn.Module):

 @add_start_docstrings(
    """
-    The GPTJ Model transformer with a language modeling head on top (linear layer with weights tied to the input
-    embeddings).
+    The GPTJ Model transformer with a language modeling head on top.
    """,
    GPTJ_START_DOCSTRING,
 )

--- a/src/transformers/models/gptj/modeling_gptj.py
+++ b/src/transformers/models/gptj/modeling_gptj.py
@@ -683,8 +683,7 @@ class GPTJModel(GPTJPreTrainedModel):

 @add_start_docstrings(
    """
-    The GPT-J Model transformer with a language modeling head on top (linear layer with weights tied to the input
-    embeddings).
+    The GPT-J Model transformer with a language modeling head on top.
    """,
    GPTJ_START_DOCSTRING,
 )