Image Feature Extraction pipeline #28216
amyeroberts
merged
25 commits into
huggingface:main
from
amyeroberts:image-fe-pipeline
amyeroberts
commented
on 2024-01-10
ydshieh
commented
on 2024-01-11
ydshieh
commented
on 2024-01-25
ydshieh
commented
on 2024-01-26
ydshieh
commented
on 2024-01-26
ydshieh
commented
on 2024-01-26
ydshieh
approved these changes
on 2024-01-26
ydshieh
commented
on 2024-02-01
amyeroberts
commented
on 2024-02-02
Draft pipeline
12a4da13
Fixup
ec40e7df
Fix docstrings
26467fcc
Update doctest
1d0a6dd5
Update pipeline_model_mapping
81303dc7
Update docstring
ba549d57
Update tests
1468c545
Update src/transformers/pipelines/image_feature_extraction.py
102422b8
Fix docstrings - review comments
466ce4e5
Remove pipeline mapping for composite vision models
1923eb16
Add to pipeline tests
9e598c9d
Remove for flava (multimodal)
f990de98
safe pil import
999e1e16
Add requirements for pipeline run
3f85b379
Account for super slow efficientnet
960b74a4
Review comments
e4098cb7
Fix tests
50c0f631
Swap order of kwargs
42853ce1
Use build_pipeline_init_args
4b059f03
Add back FE pipeline for Vilt
2c5c2fb2
Include image_processor_kwargs in docstring
95f9c585
Mark test as flaky
94945ec2
Update TODO
0a839a0a
Update tests/pipelines/test_pipelines_image_feature_extraction.py
3fd0797d
Add license header
04c109c3
Reviewers
Assignees
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub