amyeroberts ba3264b4e8 Image Feature Extraction pipeline (#28216) ... * Draft pipeline * Fixup * Fix docstrings * Update doctest * Update pipeline_model_mapping * Update docstring * Update tests * Update src/transformers/pipelines/image_feature_extraction.py Co-authored-by: Omar Sanseviero <osanseviero@gmail.com> * Fix docstrings - review comments * Remove pipeline mapping for composite vision models * Add to pipeline tests * Remove for flava (multimodal) * safe pil import * Add requirements for pipeline run * Account for super slow efficientnet * Review comments * Fix tests * Swap order of kwargs * Use build_pipeline_init_args * Add back FE pipeline for Vilt * Include image_processor_kwargs in docstring * Mark test as flaky * Update TODO * Update tests/pipelines/test_pipelines_image_feature_extraction.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> * Add license header --------- Co-authored-by: Omar Sanseviero <osanseviero@gmail.com> Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>		2024-02-05 14:50:07 +00:00
..
__init__.py	Add Donut (#18488)	2022-08-12 16:40:58 +02:00
test_image_processing_donut.py	Refactor image processor testers (#25450)	2023-08-11 11:30:18 +01:00
test_modeling_donut_swin.py	Image Feature Extraction pipeline (#28216)	2024-02-05 14:50:07 +00:00
test_processing_donut.py	Optimizes DonutProcessor token2json method for speed (#20283)	2022-11-22 10:40:59 -05:00