PR #31444 fix output data type of image classification

fix output data type of image classification #31444

amyeroberts merged 9 commits into huggingface:main from jiqing-feng:image-classification

fix output data type of image classification

fb004ef8

amyeroberts commented on 2024-06-17

add tests for low-precision pipeline

5967f786

Merge branch 'huggingface:main' into image-classification

41bd79b4

amyeroberts approved these changes on 2024-06-24

amyeroberts approved these changes on 2024-06-25

add bf16 pipeline tests

391cb629

fix bf16 tests

b759382c

Update tests/pipelines/test_pipelines_image_classification.py

6a9ca7a5

amyeroberts merged a958c4a8 into main 1 year ago

fix import

a67d088e

fix import torch

32e5cdb9

fix style

751e67fc

jiqing-feng deleted the image-classification branch 1 year ago

Reviewers

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone