accelerate

3a82b056 - Fix bf16 training with TP (#3610)

Fix bf16 training with TP (#3610) * fix * Apply style fixes --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>