Skip to content

Commit ee01736

Browse files
Merge pull request #1697 from jiqing-feng/log
fix log
2 parents ed9c8fc + ea4b59f commit ee01736

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

bitsandbytes/cextension.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -291,10 +291,9 @@ def get_native_library() -> BNBNativeLibrary:
291291
if hasattr(dll, "get_context"): # only a CUDA-built library exposes this
292292
return CudaBNBNativeLibrary(dll)
293293

294-
logger.warning(
295-
"The installed version of bitsandbytes was compiled without GPU support. "
296-
"8-bit optimizers and GPU quantization are unavailable."
297-
)
294+
# TODO: Remove this log for XPU after 8-bit optimizer is supported
295+
logger.warning("The 8-bit optimizer is not available on your device, only available on CUDA for now.")
296+
298297
return BNBNativeLibrary(dll)
299298

300299

0 commit comments

Comments
 (0)