"docs/vscode:/vscode.git/clone" did not exist on "0b5bf6abef93220fe1cf35ece99d0b54d6f00f3d"
Unverified Commit da02b403 authored by Ekagra Ranjan's avatar Ekagra Ranjan Committed by GitHub
Browse files

Add docstring for BartForCausalLM (#18795)

* add docstring for BartForCausalLM

* doc-style fic
parent 8c4a1149
......@@ -1702,6 +1702,12 @@ class BartDecoderWrapper(BartPretrainedModel):
return self.decoder(*args, **kwargs)
@add_start_docstrings(
"""
BART decoder with with a language modeling head on top (linear layer with weights tied to the input embeddings).
""",
BART_START_DOCSTRING,
)
class BartForCausalLM(BartPretrainedModel):
def __init__(self, config):
config = copy.deepcopy(config)
......
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