* Initial work

* Fix doc styler and other models
This commit is contained in:
Sylvain Gugger
2021-02-02 05:22:20 -05:00
committed by GitHub
parent 1809de5165
commit de38a6e4d2
7 changed files with 62 additions and 20 deletions

View File

@@ -412,6 +412,8 @@ RAG_FORWARD_INPUTS_DOCSTRING = r"""
Indices of input sequence tokens in the vocabulary. :class:`~transformers.RagConfig`, used to initialize
the model, specifies which generator to use, it also specifies a compatible generator tokenizer. Use that
tokenizer class to obtain the indices.
`What are input IDs? <../glossary.html#input-ids>`__
attention_mask (:obj:`torch.Tensor` of shape :obj:`(batch_size, sequence_length)`, `optional`):
Mask to avoid performing attention on padding token indices. Mask values selected in ``[0, 1]``: