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: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
  • Loading branch information
Narsil and sgugger committed Oct 29, 2021
1 parent cf5d40c commit 5cd831c
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 @@ -708,7 +708,7 @@ def __init__(self, loader, infer, params, loader_batch_size=None):
The function to apply of each element of :obj:`loader`.
params (:obj:`dict`):
The parameters passed to :obj:`infer` along with every item
loader_batch_size (:obj:`int`, :obj:`optional`):
loader_batch_size (:obj:`int`, `optional`):
If specified, the items of :obj:`loader` are supposed to come as batch, and are loader_batched here
making it roughly behave as
Expand Down

0 comments on commit 5cd831c

Please sign in to comment.