transformers
6b22a8f2
- fix bf16 issue in text classification pipeline (#30996)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
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
References
#30996 - fix bf16 issue in text classification pipeline
Author
chujiezheng
Parents
de460e28
Loading