Nicolas Patry
9f5fe63548
Adding a new align_to_words param to qa pipeline. ( #18010 )
...
* Adding a new `align_to_words` param to qa pipeline.
* Update src/transformers/pipelines/question_answering.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* Import protection.
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-08-09 18:50:02 +02:00
Nicolas Patry
a4562552eb
[DX fix] Fixing QA pipeline streaming a dataset. ( #18516 )
...
* [DX fix] Fixing QA pipeline streaming a dataset.
QuestionAnsweringArgumentHandler would iterate over the whole dataset
effectively killing all properties of the pipeline.
This restores nice properties when using `Dataset` or `Generator` since
those are meant to be consumed lazily.
* Handling TF better.
2022-08-08 14:25:56 +02:00
Nicolas Patry
2b282296f1
Adding batch_size test to QA pipeline. ( #17330 )
2022-05-19 14:28:12 -04:00
Sylvain Gugger
afe5d42d8d
Black preview ( #17217 )
...
* Black preview
* Fixup too!
* Fix check copies
* Use the same version as the CI
* Bump black
2022-05-12 16:25:55 -04:00
Nicolas Patry
6d80c92c77
LogSumExp trick question_answering pipeline. ( #17143 )
...
* LogSumExp trick `question_answering` pipeline.
* Adding a failing test.
2022-05-10 10:03:55 +02:00
Nicolas Patry
6620f60c0a
Long QuestionAnsweringPipeline fix. ( #16778 )
...
* Temporary commit witht the long QA fix.
* Adding slow tests covering this fix.
* Removing fast test as it doesn't fail anyway.
2022-04-21 09:59:25 +02:00
Lysandre Debut
29c10a41d0
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
...
* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>
2022-02-23 15:46:28 -05:00
