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
..
2023-09-05 12:27:20 +01:00
2024-02-01 09:16:16 -08:00
2023-11-09 12:19:31 +00:00
2023-06-20 18:07:47 -04:00
2023-06-20 18:07:47 -04:00
2024-01-24 08:31:28 -08:00
2023-10-16 09:52:29 +02:00
2023-06-20 18:07:47 -04:00
2023-06-20 18:07:47 -04:00
2023-10-12 10:48:38 +02:00
2023-07-13 16:56:53 +01:00
2023-06-20 18:07:47 -04:00
2023-06-20 18:07:47 -04:00
2023-10-30 09:39:14 -07:00
2024-02-05 14:50:07 +00:00
2023-11-06 19:45:03 +00:00
2024-02-02 08:22:18 +01:00
2023-06-20 18:07:47 -04:00
2023-11-06 15:11:00 +01:00
2023-12-20 10:37:23 -08:00