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

..

internal

Generate: consolidate output classes (#28494)

2024-01-15 17:04:08 +00:00

main_classes

Image Feature Extraction pipeline (#28216)

2024-02-05 14:50:07 +00:00

model_doc

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

tasks

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

_config.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00

_redirects.yml

Extended semantic segmentation to image segmentation (#27039)

2023-11-23 15:58:21 +00:00

_toctree.yml

[HfQuantizer] Move it to "Developper guides" (#28768)

2024-01-30 07:20:20 +01:00

accelerate.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

add_new_model.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

add_new_pipeline.md

Fix broken link on page (#28451)

2024-01-11 09:26:13 -08:00

add_tensorflow_model.md

Remove utils/documentation_tests.txt (#26213)

2023-09-18 13:33:01 +02:00

attention.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

autoclass_tutorial.md

[docs] Backbone (#28739)

2024-02-01 09:16:16 -08:00

benchmarks.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

bertology.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

big_models.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

chat_templating.md

Add tip on setting tokenizer attributes (#28764)

2024-02-01 14:44:58 +00:00

community.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

contributing.md

Enable doc in Spanish (#16518)

2022-04-04 10:25:46 -04:00

create_a_model.md

[docs] Backbone (#28739)

2024-02-01 09:16:16 -08:00

custom_models.md

Add config tip to custom model docs (#28601)

2024-01-22 13:46:04 +00:00

custom_tools.md

[doc] Always call it Agents for consistency (#25958)

2023-09-05 12:27:20 +01:00

debugging.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

deepspeed.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

fast_tokenizers.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

fsdp.md

[docs] Trainer docs (#28145)

2023-12-20 10:37:23 -08:00

generation_strategies.md

Generate: fix speculative decoding (#28166)

2023-12-20 18:55:35 +00:00

glossary.md

[Doc] Spanish translation of glossary.md (#27958)

2023-12-13 09:21:59 -08:00

hf_quantizer.md

[docs] HfQuantizer (#28820)

2024-02-02 08:22:18 +01:00

hpo_train.md

Remove-auth-token (#27060)

2023-11-13 14:20:54 +01:00

index.md

Flax mistral (#26943)

2024-01-31 14:19:02 +01:00

installation.md

README: install transformers from conda-forge channel (#28313)

2024-01-04 09:36:16 -08:00

llm_tutorial_optimization.md

F.scaled_dot_product_attention support (#26572)

2023-12-09 05:38:14 +09:00

llm_tutorial.md

Generate: All logits processors are documented and have examples (#27796)

2023-12-07 15:11:35 +00:00

model_memory_anatomy.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

model_sharing.md

Fix typos (#25936)

2023-09-04 11:15:12 +01:00

model_summary.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

multilingual.md

Fix typo in example code (#25583)

2023-08-18 07:58:59 +02:00

notebooks.md

Enable doc in Spanish (#16518)

2022-04-04 10:25:46 -04:00

pad_truncation.md

[Doc] Spanish translation of pad_truncation.md (#27890)

2023-12-08 10:32:18 -08:00

peft.md

[Peft] modules_to_save support for peft integration (#27466)

2023-11-14 10:32:57 +01:00

perf_hardware.md

docs: replace torch.distributed.run by torchrun (#27528)

2023-11-27 16:26:33 +00:00

perf_infer_cpu.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

perf_infer_gpu_one.md

Add qwen2 (#28436)

2024-01-17 16:02:22 +01:00

perf_torch_compile.md

Fix rendering for torch.compile() docs (#25432)

2023-08-10 13:25:00 +02:00

perf_train_cpu_many.md

Doc (#28431)

2024-01-11 08:55:48 -08:00

perf_train_cpu.md

improve efficient training on CPU documentation (#28646)

2024-01-24 09:07:13 -08:00

perf_train_gpu_many.md

[docs] Improve visualization for vertical parallelism (#28583)

2024-01-25 17:55:11 +00:00

perf_train_gpu_one.md

Improving Training Performance and Scalability Documentation (#28497)

2024-01-16 11:30:26 +01:00

perf_train_special.md

[docs] MPS (#28016)

2023-12-15 13:17:29 -08:00

perf_train_tpu_tf.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

performance.md

[docs] Update CPU/GPU inference docs (#26881)

2023-10-31 09:44:51 -07:00

perplexity.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

philosophy.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

pipeline_tutorial.md

[ASR Pipe] Improve docs and error messages (#26476)

2023-09-29 18:32:37 +01:00

pipeline_webserver.md

Suggestions on Pipeline_webserver (#25570)

2023-08-18 10:17:44 +02:00

pr_checks.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

preprocessing.md

[docs] Update preprocessing.md (#28719)

2024-01-26 11:58:57 +00:00

quantization.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

quicktour.md

[TYPO] fix typo/format in quicktour.md (#25519)

2023-08-16 08:03:23 +02:00

run_scripts.md

docs: replace torch.distributed.run by torchrun (#27528)

2023-11-27 16:26:33 +00:00

sagemaker.md

[docs] fixed links with 404 (#27327)

2023-11-06 19:45:03 +00:00

serialization.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

task_summary.md

[Doc] Fix token link in What 🤗 Transformers can do (#28123)

2023-12-18 15:06:54 -08:00

tasks_explained.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

testing.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

tf_xla.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

tflite.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

tokenizer_summary.md

[Docs] Fix spelling and grammar mistakes (#28825)

2024-02-02 08:45:00 +01:00

torchscript.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00

trainer.md

[docs] Trainer docs (#28145)

2023-12-20 10:37:23 -08:00

training.md

Fix semantic error in evaluation section (#27675)

2023-11-24 12:41:16 +01:00

transformers_agents.md

[doc] Always call it Agents for consistency (#25958)

2023-09-05 12:27:20 +01:00

troubleshooting.md

Migrate doc files to Markdown. (#24376)

2023-06-20 18:07:47 -04:00