jiqing-feng 49a0bef4c1 enable low-precision pipeline (#31625) ...

* enable low-precision pipeline

* fix parameter for ASR

* reformat

* fix asr bug

* fix bug for zero-shot

* add dtype check

* rm useless comments

* add np.float16 check

* Update src/transformers/pipelines/image_classification.py

Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>

* Update src/transformers/pipelines/token_classification.py

Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>

* fix comments

* fix asr check

* make fixup

* No more need for is_torch_available()

---------

Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
Co-authored-by: Matt <Rocketknight1@users.noreply.github.com>
Co-authored-by: Matt <rocketknight1@gmail.com>

2024-09-20 16:43:30 -07: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

Remove trust_remote_code when loading Libri Dummy (#31748)

2024-07-23 14:54:38 +08:00

test_pipelines_automatic_speech_recognition.py

enable low-precision pipeline (#31625)

2024-09-20 16:43:30 -07:00

test_pipelines_common.py

Pipeline: no side-effects on model.config and model.generation_config 🔫 (#33480)

2024-09-18 15:43:06 +01:00

test_pipelines_depth_estimation.py

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

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

test_pipelines_document_question_answering.py

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

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

test_pipelines_feature_extraction.py

fix: Fixed raising TypeError instead of ValueError for invalid type (#32111)

2024-07-22 17:46:17 +01:00

test_pipelines_fill_mask.py

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

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

test_pipelines_image_classification.py

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

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

test_pipelines_image_feature_extraction.py

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

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

test_pipelines_image_segmentation.py

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

2024-07-05 17:21:50 +01: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

Avoid failure TFBlipModelTest::test_pipeline_image_to_text (#31827)

2024-07-08 13:49:21 +02:00

test_pipelines_mask_generation.py

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

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

test_pipelines_object_detection.py

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

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

test_pipelines_question_answering.py

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

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

test_pipelines_summarization.py

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

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

test_pipelines_table_question_answering.py

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

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

test_pipelines_text2text_generation.py

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

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

test_pipelines_text_classification.py

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

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

test_pipelines_text_generation.py

Compile compatibilty for decoder-only models (#32617)

2024-09-09 10:59:04 +02:00

test_pipelines_text_to_audio.py

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

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

test_pipelines_token_classification.py

enable low-precision pipeline (#31625)

2024-09-20 16:43:30 -07:00

test_pipelines_translation.py

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

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

test_pipelines_video_classification.py

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

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

test_pipelines_visual_question_answering.py

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

2024-07-05 17:21:50 +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

🚨 Add Blip2ForImageTextRetrieval (#29261)

2024-08-27 18:50:27 +01:00

test_pipelines_zero_shot_object_detection.py

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

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

test_pipelines_zero_shot.py

enable low-precision pipeline (#31625)

2024-09-20 16:43:30 -07:00