Skip to content

Commit

Permalink
Update src/transformers/pipelines/base.py
Browse files Browse the repository at this point in the history
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
  • Loading branch information
amyeroberts and pcuenca authored Jan 18, 2024
1 parent c96a5b9 commit 29dd880
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/pipelines/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -770,7 +770,7 @@ def build_pipeline_init_args(
workers to be used.
batch_size (`int`, *optional*, defaults to 1):
When the pipeline will use *DataLoader* (when passing a dataset, on GPU for a Pytorch model), the size of
the batch to use, for inference this is not always beneficial, please read [Batching with
the batch to use. For inference this is not always beneficial, please read [Batching with
pipelines](https://huggingface.co/docs/transformers/main_classes/pipelines#pipeline-batching).
args_parser ([`~pipelines.ArgumentHandler`], *optional*):
Reference to the object in charge of parsing supplied pipeline parameters.
Expand Down

0 comments on commit 29dd880

Please sign in to comment.