Update main.py for return cuda118

This commit is contained in:
青龍聖者@bdsqlsz
2023-04-25 11:38:03 +08:00
committed by GitHub
parent fa6d7485ac
commit 943509dae4

View File

@@ -410,6 +410,7 @@ def evaluate_cuda_setup():
print(('Welcome to bitsandbytes. For bug reports, please run\n\npython -m bitsandbytes\n\n'),
('and submit this information together with your error trace to: https://github.com/TimDettmers/bitsandbytes/issues'))
print('='*80)
return 'libbitsandbytes_cuda118.dll', None, None, None, None
if not torch.cuda.is_available(): return 'libbitsandbytes_cpu'+SHARED_LIB_EXTENSION, None, None, None, None
cuda_setup = CUDASetup.get_instance()