SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[docs] fix xref to PreTrainedModel.generate (#11049)

Browse Source 
* fix xref to generate

* do the same for search methods

* style

* style
This commit is contained in:
Stas Bekman
2021-06-02 09:21:05 -07:00
committed by GitHub
parent 123b597f5d
commit d406a2729a
6 changed files with 37 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/model_doc/bart.rst
View File

@@ -61,7 +61,7 @@ Implementation Notes
- Model predictions are intended to be identical to the original implementation when
:obj:`force_bos_token_to_be_generated=True`. This only works, however, if the string you pass to
:func:`fairseq.encode` starts with a space.
- :meth:`~transformers.BartForConditionalGeneration.generate` should be used for conditional generation tasks like
- :meth:`~transformers.generation_utils.GenerationMixin.generate` should be used for conditional generation tasks like
summarization, see the example in that docstrings.
- Models that load the `facebook/bart-large-cnn` weights will not have a :obj:`mask_token_id`, or be able to perform
mask-filling tasks.