Skip to content

Commit 35dbb1f

Browse files
committed
Fix bitsandbytes import error when CUDA is unavailable
1 parent 486488b commit 35dbb1f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/cuda_setup/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ def evaluate_cuda_setup():
330330
cuda_setup.add_log_entry(('Welcome to bitsandbytes. For bug reports, please run\n\npython -m bitsandbytes\n\n'),
331331
('and submit this information together with your error trace to: https://github.com/TimDettmers/bitsandbytes/issues'))
332332
cuda_setup.add_log_entry('='*80)
333-
if not torch.cuda.is_available(): return 'libbitsandbytes_cpu.so', None, None, None, None
333+
if not torch.cuda.is_available(): return 'libbitsandbytes_cpu.so', None, None, None
334334

335335
cudart_path = determine_cuda_runtime_lib_path()
336336
ccs = get_compute_capabilities()

0 commit comments

Comments
 (0)