transformers
6b22a8f2 - fix bf16 issue in text classification pipeline (#30996)

Commit
1 year ago
fix bf16 issue in text classification pipeline (#30996) * fix logits dtype * Add bf16/fp16 tests for text_classification pipeline * Update test_pipelines_text_classification.py * fix * fix
Author
Parents
Loading