qlora icon indicating copy to clipboard operation
qlora copied to clipboard
verified publisher icon artidoro

Enhance GPU `bfloat16` support check

Open AlpinDale opened this issue 2 years ago • 0 comments

This PR updates the existing code to check GPU support for bfloat16 using torch.cuda.is_bf16_supported() function, which returns a boolean value. This should provide a more robust approach than using torch.cuda.get_device_capability().

AlpinDale avatar Jun 22 '23 16:06 AlpinDale