Unverified Commit 0acd84f7 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[GPTJ] fix docs (#15558)

parent 87d08afb
...@@ -668,8 +668,7 @@ class FlaxGPTJForCausalLMModule(nn.Module): ...@@ -668,8 +668,7 @@ class FlaxGPTJForCausalLMModule(nn.Module):
@add_start_docstrings( @add_start_docstrings(
""" """
The GPTJ Model transformer with a language modeling head on top (linear layer with weights tied to the input The GPTJ Model transformer with a language modeling head on top.
embeddings).
""", """,
GPTJ_START_DOCSTRING, GPTJ_START_DOCSTRING,
) )
......
...@@ -683,8 +683,7 @@ class GPTJModel(GPTJPreTrainedModel): ...@@ -683,8 +683,7 @@ class GPTJModel(GPTJPreTrainedModel):
@add_start_docstrings( @add_start_docstrings(
""" """
The GPT-J Model transformer with a language modeling head on top (linear layer with weights tied to the input The GPT-J Model transformer with a language modeling head on top.
embeddings).
""", """,
GPTJ_START_DOCSTRING, GPTJ_START_DOCSTRING,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment