Code

Image Feature Extraction pipeline#28216

Code

Merged

amyeroberts merged 25 commits intohuggingface:mainhuggingface/transformers:mainfrom amyeroberts:image-fe-pipelineamyeroberts/transformers:image-fe-pipelineCopy head branch name to clipboardFeb 5, 2024

+387-53

Commits

Commits on Feb 5, 2024

• Draft pipeline

  amyeroberts

  committed
  12a4da1

• Fixup

  amyeroberts

  committed
  ec40e7d

• Fix docstrings

  amyeroberts

  committed
  26467fc

• Update doctest

  amyeroberts

  committed
  1d0a6dd

• Update pipeline_model_mapping

  amyeroberts

  committed
  81303dc

• Update docstring

  amyeroberts

  committed
  ba549d5

• Update tests

  amyeroberts

  committed
  1468c54

• Update src/transformers/pipelines/image_feature_extraction.py

  Show description for 102422b
  

  amyeroberts

  and

  osanseviero

  committed
  102422b

• Fix docstrings - review comments

  amyeroberts

  committed
  466ce4e

• Remove pipeline mapping for composite vision models

  amyeroberts

  committed
  1923eb1

• Add to pipeline tests

  amyeroberts

  committed
  9e598c9

• Remove for flava (multimodal)

  amyeroberts

  committed
  f990de9

• safe pil import

  amyeroberts

  committed
  999e1e1

• Add requirements for pipeline run

  amyeroberts

  committed
  3f85b37

• Account for super slow efficientnet

  amyeroberts

  committed
  960b74a

• Review comments

  amyeroberts

  committed
  e4098cb

• Fix tests

  amyeroberts

  committed
  50c0f63

• Swap order of kwargs

  amyeroberts

  committed
  42853ce

• Use build_pipeline_init_args

  amyeroberts

  committed
  4b059f0

• Add back FE pipeline for Vilt

  amyeroberts

  committed
  2c5c2fb

• Include image_processor_kwargs in docstring

  amyeroberts

  committed
  95f9c58

• Mark test as flaky

  amyeroberts

  committed
  94945ec

• Update TODO

  amyeroberts

  committed
  0a839a0

• Update tests/pipelines/test_pipelines_image_feature_extraction.py

  Show description for 3fd0797
  

  amyeroberts

  and

  ArthurZucker

  committed
  3fd0797

• Add license header

  amyeroberts

  committed
  04c109c