mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-01 02:31:11 +06: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> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| test_pipelines_audio_classification.py | ||
| test_pipelines_automatic_speech_recognition.py | ||
| test_pipelines_common.py | ||
| test_pipelines_conversational.py | ||
| test_pipelines_depth_estimation.py | ||
| test_pipelines_document_question_answering.py | ||
| test_pipelines_feature_extraction.py | ||
| test_pipelines_fill_mask.py | ||
| test_pipelines_image_classification.py | ||
| test_pipelines_image_feature_extraction.py | ||
| test_pipelines_image_segmentation.py | ||
| test_pipelines_image_to_image.py | ||
| test_pipelines_image_to_text.py | ||
| test_pipelines_mask_generation.py | ||
| test_pipelines_object_detection.py | ||
| test_pipelines_question_answering.py | ||
| test_pipelines_summarization.py | ||
| test_pipelines_table_question_answering.py | ||
| test_pipelines_text_classification.py | ||
| test_pipelines_text_generation.py | ||
| test_pipelines_text_to_audio.py | ||
| test_pipelines_text2text_generation.py | ||
| test_pipelines_token_classification.py | ||
| test_pipelines_translation.py | ||
| test_pipelines_video_classification.py | ||
| test_pipelines_visual_question_answering.py | ||
| test_pipelines_zero_shot_audio_classification.py | ||
| test_pipelines_zero_shot_image_classification.py | ||
| test_pipelines_zero_shot_object_detection.py | ||
| test_pipelines_zero_shot.py | ||