jiqing-feng a958c4a801 fix output data type of image classification (#31444) ...

* fix output data type of image classification

* add tests for low-precision pipeline

* add bf16 pipeline tests

* fix bf16 tests

* Update tests/pipelines/test_pipelines_image_classification.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* fix import

* fix import torch

* fix style

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

2024-06-25 11:14:39 +01: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

Pass datasets trust_remote_code (#31406)

2024-06-17 17:29:13 +01:00

test_pipelines_automatic_speech_recognition.py

Pass datasets trust_remote_code (#31406)

2024-06-17 17:29:13 +01:00

test_pipelines_common.py

Pass datasets trust_remote_code (#31406)

2024-06-17 17:29:13 +01:00

test_pipelines_depth_estimation.py

Update fixtures-image-utils (#28080)

2023-12-15 16:58:36 +00:00

test_pipelines_document_question_answering.py

Fix all torch pipeline failures except one (#30290)

2024-04-18 10:35:43 +02:00

test_pipelines_feature_extraction.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_fill_mask.py

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

test_pipelines_image_classification.py

fix output data type of image classification (#31444)

2024-06-25 11:14:39 +01:00

test_pipelines_image_feature_extraction.py

[pipeline] Add pool option to image feature extraction pipeline (#28985)

2024-02-20 20:22:08 +00:00

test_pipelines_image_segmentation.py

Pass datasets trust_remote_code (#31406)

2024-06-17 17:29:13 +01:00

test_pipelines_image_to_image.py

Add image to image pipeline (#25393)

2023-09-22 19:53:55 +03:00

test_pipelines_image_to_text.py

[tests] fix the wrong output in ImageToTextPipelineTests.test_conditional_generation_llava (#29975)

2024-04-01 13:08:39 +02:00

test_pipelines_mask_generation.py

Set usedforsecurity=False in hashlib methods (FIPS compliance) (#27483)

2023-11-16 14:29:53 +00:00

test_pipelines_object_detection.py

Update fixtures-image-utils (#28080)

2023-12-15 16:58:36 +00:00

test_pipelines_question_answering.py

[Styling] stylify using ruff (#27144)

2023-11-16 17:43:19 +01:00

test_pipelines_summarization.py

Proper build() methods for TF (#27794)

2023-12-14 15:17:30 +00:00

test_pipelines_table_question_answering.py

Skip some TQAPipelineTests tests in past CI (#24267)

2023-06-14 14:25:24 +02:00

test_pipelines_text2text_generation.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_text_classification.py

Fix pipeline tests - torch imports (#31227)

2024-06-04 12:30:23 +01:00

test_pipelines_text_generation.py

Fix bug about add_special_tokens and so on (#31496)

2024-06-24 14:05:16 +01:00

test_pipelines_text_to_audio.py

if output is tuple like facebook/hf-seamless-m4t-medium, waveform is … (#29722)

2024-04-05 09:26:44 +02:00

test_pipelines_token_classification.py

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00

test_pipelines_translation.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_video_classification.py

Replace 'decord' with 'av' in VideoClassificationPipeline (#29747)

2024-03-26 10:12:24 +00:00

test_pipelines_visual_question_answering.py

Pipeline VQA: Add support for list of images and questions as pipeline input (#31217)

2024-06-06 14:50:45 +01:00

test_pipelines_zero_shot_audio_classification.py

Fix all torch pipeline failures except one (#30290)

2024-04-18 10:35:43 +02:00

test_pipelines_zero_shot_image_classification.py

Add SigLIP (#26522)

2024-01-08 18:17:16 +01:00

test_pipelines_zero_shot_object_detection.py

Mark pipeline tests to skip them easily (#21887)

2023-03-02 10:55:36 -05:00

test_pipelines_zero_shot.py

Update all references to canonical models (#29001)

2024-02-16 08:16:58 +01:00