Skip to content

Commit

Permalink
Improve docstring.
Browse files Browse the repository at this point in the history
  • Loading branch information
Narsil committed Mar 5, 2021
1 parent 5eefa4b commit 85f69b4
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/transformers/pipelines/text2text_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -294,9 +294,11 @@ def __call__(
clean_up_tokenization_spaces (:obj:`bool`, `optional`, defaults to :obj:`False`):
Whether or not to clean up the potential extra spaces in the text output.
src_lang (:obj:`str`, `optional`, defaults to :obj:`None`):
The language of the input. Might be required for multilingual models
The language of the input. Might be required for multilingual models. Will not have any effect
for single pair translation models
tgt_lang (:obj:`str`, `optional`, defaults to :obj:`None`):
The language of the desired output. Might be required for multilingual models
The language of the desired output. Might be required for multilingual models. Will not have any effect
for single pair translation models
generate_kwargs:
Additional keyword arguments to pass along to the generate method of the model (see the generate method
corresponding to your framework `here <./model.html#generative-models>`__).
Expand Down

0 comments on commit 85f69b4

Please sign in to comment.