🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
Fix XLA fp16 and bf16 error checking (#18913)
* Fix XLA fp16 and bf16 error checking * Update src/transformers/training_args.py Co-authored-by: Sylvain Gugger <[email protected]> Co-authored-by: Sylvain Gugger <[email protected]>
Y
Yanming Wang committed
639422187157bb19339e0ecdd6eeff7e1011a801
Parent: 6690ba3
Committed by GitHub <[email protected]>
on 9/7/2022, 7:45:17 PM