Skip to content

Commit

Permalink
Pipeline: fix bad generation kwargs docs (#33205)
Browse files Browse the repository at this point in the history
fix link
  • Loading branch information
gante authored Aug 30, 2024
1 parent c409cd8 commit b127fb8
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/transformers/pipelines/text2text_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def __call__(self, *args, **kwargs):
max_length instead of throwing an error down the line.
generate_kwargs:
Additional keyword arguments to pass along to the generate method of the model (see the generate method
corresponding to your framework [here](./main_classes/text_generation)).
corresponding to your framework [here](./text_generation)).
Return:
A list or a list of list of `dict`: Each result comes as a dictionary with the following keys:
Expand Down Expand Up @@ -257,7 +257,7 @@ def __call__(self, *args, **kwargs):
Whether or not to clean up the potential extra spaces in the text output.
generate_kwargs:
Additional keyword arguments to pass along to the generate method of the model (see the generate method
corresponding to your framework [here](./main_classes/text_generation)).
corresponding to your framework [here](./text_generation)).
Return:
A list or a list of list of `dict`: Each result comes as a dictionary with the following keys:
Expand Down Expand Up @@ -359,7 +359,7 @@ def __call__(self, *args, **kwargs):
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](./main_classes/text_generation)).
corresponding to your framework [here](./text_generation)).
Return:
A list or a list of list of `dict`: Each result comes as a dictionary with the following keys:
Expand Down

0 comments on commit b127fb8

Please sign in to comment.