transformers
ec9b18f6 - Fix --bf16 option support for Neuron after PR #22300 (#22307)

Commit
2 years ago
Fix --bf16 option support for Neuron after PR #22300 (#22307) This PR fixes the "RuntimeError: No CUDA GPUs are available" when running with --bf16 option on Neuron. Related PRs: https://github.com/huggingface/transformers/pull/20684 https://github.com/huggingface/transformers/pull/22300
Author
Parents
Loading