tests/test_pipelines.py · 0a6cbea0a5e130ea6f935588f2a66b89b8aa9684 · chenpangpang / transformers

Rewritten batch support in pipelines. (#4154) · 0a6cbea0

Funtowicz Morgan authored May 07, 2020



* Rewritten batch support in pipelines.
Signed-off-by: Morgan Funtowicz <morgan@huggingface.co>

* Fix imports sorting 🔧

Signed-off-by: Morgan Funtowicz <morgan@huggingface.co>

* Set pad_to_max_length=True by default on Pipeline.

* Set pad_to_max_length=False for generation pipelines.

Most of generation models doesn't have padding token.

* Address @joeddav review comment: Uniformized *args.
Signed-off-by: Morgan Funtowicz <morgan@huggingface.co>

* Address @joeddav review comment: Uniformized *args (second).
Signed-off-by: Morgan Funtowicz <morgan@huggingface.co>

0a6cbea0

test_pipelines.py 18.8 KB

Replace test_pipelines.py