Skip to content

Commit

Permalink
[pipeline doc] fix api (huggingface#15660)
Browse files Browse the repository at this point in the history
* [pipeline doc] fix api

* remove duplicate
  • Loading branch information
stas00 authored and ManuelFay committed Mar 31, 2022
1 parent 171dc02 commit ac12e19
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/transformers/pipelines/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -444,16 +444,15 @@ def pipeline(
If no framework is specified, will default to the one currently installed. If no framework is specified and
both frameworks are installed, will default to the framework of the `model`, or to PyTorch if no model is
provided.
revision(`str`, *optional*, defaults to `"main"`):
revision (`str`, *optional*, defaults to `"main"`):
When passing a task name or a string model identifier: The specific model version to use. It can be a
branch name, a tag name, or a commit id, since we use a git-based system for storing models and other
artifacts on huggingface.co, so `revision` can be any identifier allowed by git.
use_fast (`bool`, *optional*, defaults to `True`):
Whether or not to use a Fast tokenizer if possible (a [`PreTrainedTokenizerFast`]).
use_auth_token (`str` or *bool*, *optional*):
The token to use as HTTP bearer authorization for remote files. If `True`, will use the token generated
when running `transformers-cli login` (stored in `~/.huggingface`). revision(`str`, *optional*, defaults to
`"main"`):
when running `transformers-cli login` (stored in `~/.huggingface`).
model_kwargs:
Additional dictionary of keyword arguments passed along to the model's `from_pretrained(...,
**model_kwargs)` function.
Expand Down

0 comments on commit ac12e19

Please sign in to comment.