Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
src/transformers/models/bart/modeling_bart.py
View file @ da02b403
...@@ -1702,6 +1702,12 @@ class BartDecoderWrapper(BartPretrainedModel): ...@@ -1702,6 +1702,12 @@ class BartDecoderWrapper(BartPretrainedModel):
return self.decoder(*args, **kwargs) 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): class BartForCausalLM(BartPretrainedModel):
def __init__(self, config): def __init__(self, config):
config = copy.deepcopy(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