"...composable_kernel_rocm.git" did not exist on "a6ef5c391e893c423a34c7a80d99009ddf762af4"
Unverified Commit 965cf677 authored by Alvaro Bartolome's avatar Alvaro Bartolome Committed by GitHub
Browse files

Fix `TextGenerationPipeline.__call__` docstring (#29491)

parent 19fb1e22
...@@ -194,7 +194,7 @@ class TextGenerationPipeline(Pipeline): ...@@ -194,7 +194,7 @@ class TextGenerationPipeline(Pipeline):
Complete the prompt(s) given as inputs. Complete the prompt(s) given as inputs.
Args: Args:
args (`str` or `List[str]`): text_inputs (`str` or `List[str]`):
One or several prompts (or one list of prompts) to complete. One or several prompts (or one list of prompts) to complete.
return_tensors (`bool`, *optional*, defaults to `False`): 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 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): ...@@ -217,8 +217,7 @@ class TextGenerationPipeline(Pipeline):
- `None` : default strategy where nothing in particular happens - `None` : default strategy where nothing in particular happens
- `"hole"`: Truncates left of input, and leaves a gap wide enough to let generation happen (might - `"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) truncate a lot of the prompt and not suitable when generation exceed the model capacity)
generate_kwargs (`dict`, *optional*):
generate_kwargs:
Additional keyword arguments to pass along to the generate method of the model (see the generate method 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)). corresponding to your framework [here](./model#generative-models)).
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment