Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
965cf677
Unverified
Commit
965cf677
authored
Mar 06, 2024
by
Alvaro Bartolome
Committed by
GitHub
Mar 06, 2024
Browse files
Fix `TextGenerationPipeline.__call__` docstring (#29491)
parent
19fb1e22
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
src/transformers/pipelines/text_generation.py
src/transformers/pipelines/text_generation.py
+2
-3
No files found.
src/transformers/pipelines/text_generation.py
View file @
965cf677
...
...
@@ -194,7 +194,7 @@ class TextGenerationPipeline(Pipeline):
Complete the prompt(s) given as inputs.
Args:
arg
s (`str` or `List[str]`):
text_input
s (`str` or `List[str]`):
One or several prompts (or one list of prompts) to complete.
return_tensors (`bool`, *optional*, defaults to `False`):
Whether or not to return the tensors of predictions (as token indices) in the outputs. If set to
...
...
@@ -217,8 +217,7 @@ class TextGenerationPipeline(Pipeline):
- `None` : default strategy where nothing in particular happens
- `"hole"`: Truncates left of input, and leaves a gap wide enough to let generation happen (might
truncate a lot of the prompt and not suitable when generation exceed the model capacity)
generate_kwargs:
generate_kwargs (`dict`, *optional*):
Additional keyword arguments to pass along to the generate method of the model (see the generate method
corresponding to your framework [here](./model#generative-models)).
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment