tests/pipelines/test_pipelines_image_classification.py · a958c4a801a0db655981678b050eaa855289405c · chenpangpang / transformers

"tests/models/cvt/test_modeling_tf_cvt.py" did not exist on "0a057201a96565df29984d716f660fd8d634329a"

fix output data type of image classification (#31444) · a958c4a8

jiqing-feng authored Jun 25, 2024



* 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>

a958c4a8

test_pipelines_image_classification.py 11 KB

Replace test_pipelines_image_classification.py