tests/pipelines/test_pipelines_text_classification.py · 6b22a8f2d8ded6b2a681e1c078b0e1abf27d045c · chenpangpang / transformers

Chujie Zheng authored Jun 04, 2024

* fix logits dtype

* Add bf16/fp16 tests for text_classification pipeline

* Update test_pipelines_text_classification.py

* fix

* fix

6b22a8f2

test_pipelines_text_classification.py 9.26 KB