mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-15 21:01:19 +00:00
ba3264b4e8
* 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> |
||
|---|---|---|
| .. | ||
| agent.md | ||
| backbones.md | ||
| callback.md | ||
| configuration.md | ||
| data_collator.md | ||
| deepspeed.md | ||
| feature_extractor.md | ||
| image_processor.md | ||
| keras_callbacks.md | ||
| logging.md | ||
| model.md | ||
| onnx.md | ||
| optimizer_schedules.md | ||
| output.md | ||
| pipelines.md | ||
| processors.md | ||
| quantization.md | ||
| text_generation.md | ||
| tokenizer.md | ||
| trainer.md | ||