transformers
021006e1 - Fix removing "cpu" from frozenset in bitsandbytes.py to allow better ROCm support. (#36975)

Commit
315 days ago
Fix removing "cpu" from frozenset in bitsandbytes.py to allow better ROCm support. (#36975) * Fix removing "cpu" from frozenset in bitsandbytes.py to allow better ROCm support. Related to https://github.com/bitsandbytes-foundation/bitsandbytes/issues/1573 and https://github.com/huggingface/transformers/issues/36949 , this resolves a bug in allowing ROCm/HIP support in bitsandbytes. * Related to bitsandbytes-foundation/bitsandbytes#1573 and huggingface#36949 , this resolves a bug in the biteandbytes integration, allowing ROCm/HIP support in bitsandbytes. --------- Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
Author
Parents
Loading