Yoni Gozlan 9f51dc2535 Add support for post-processing kwargs in image-text-to-text pipeline (#35374) ...

* fix error and improve pipeline

* add processing_kwargs to apply_chat_template

* change default post_process kwarg to args

* Fix slow tests

* fix copies

2025-02-18 17:43:36 -05:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_pipelines_audio_classification.py

Move audio top_k tests to the right file and add slow decorator (#36072)

2025-02-07 14:32:30 +00:00

test_pipelines_automatic_speech_recognition.py

Pipeline: simple API for assisted generation (#34504)

2025-01-08 17:08:02 +00:00

test_pipelines_common.py

docs: fix return type annotation of get_default_model_revision (#35982)

2025-02-13 11:59:15 +01:00

test_pipelines_depth_estimation.py

Add post_process_depth_estimation to image processors and support ZoeDepth's inference intricacies (#32550)

2024-10-22 15:50:54 +02:00

test_pipelines_document_question_answering.py

fix document qa bf16 pipeline (#35456)

2025-01-20 16:18:07 +00:00

test_pipelines_feature_extraction.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_fill_mask.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_classification.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_feature_extraction.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_segmentation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_image_text_to_text.py

Add support for post-processing kwargs in image-text-to-text pipeline (#35374)

2025-02-18 17:43:36 -05:00

test_pipelines_image_to_image.py

Allow FP16 or other precision inference for Pipelines (#31342)

2024-07-05 17:21:50 +01:00

test_pipelines_image_to_text.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_mask_generation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_object_detection.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_question_answering.py

enable QA bf16 pipeline (#34483)

2024-10-31 12:55:53 +00:00

test_pipelines_summarization.py

Avoid check expected exception when it is on CUDA (#34408)

2024-10-25 17:14:07 +02:00

test_pipelines_table_question_answering.py

Aurevoir PyTorch 1 (#35358)

2024-12-20 14:36:31 +01:00

test_pipelines_text2text_generation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_text_classification.py

Fix default behaviour in TextClassificationPipeline for regression problem type (#34066)

2024-10-15 13:06:20 +01:00

test_pipelines_text_generation.py

Output dicts support in text generation pipeline (#35092)

2025-01-29 14:44:46 +00:00

test_pipelines_text_to_audio.py

Fix seamless TTS generate (#34968)

2024-12-11 15:38:42 +01:00

test_pipelines_token_classification.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_translation.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_video_classification.py

Fix test_pipelines_video_classification that was always failing (#35842)

2025-01-23 19:22:32 +01:00

test_pipelines_visual_question_answering.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

test_pipelines_zero_shot_audio_classification.py

Allow FP16 or other precision inference for Pipelines (#31342)

2024-07-05 17:21:50 +01:00

test_pipelines_zero_shot_image_classification.py

Image pipelines spec compliance (#33899)

2024-10-08 13:34:28 +01:00

test_pipelines_zero_shot_object_detection.py

Add image text to text pipeline (#34170)

2024-10-31 15:48:11 -04:00

test_pipelines_zero_shot.py

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00