lib methods can now safely be assigned, no more cryptic errors on missing lib

bitsandbytes/functional.py

@@ -19,9 +19,8 @@ from .cextension import lib
 
 name2qmap = {}
 
-if lib and lib.compiled_with_cuda:
-    """C FUNCTIONS FOR OPTIMIZERS"""
-    str2optimizer32bit = {
+"""C FUNCTIONS FOR OPTIMIZERS"""
+str2optimizer32bit = {
     "adam": (
         lib.cadam32bit_grad_fp32,
         lib.cadam32bit_grad_fp16,
@@ -54,9 +53,9 @@ if lib and lib.compiled_with_cuda:
         lib.cademamix32bit_grad_fp16,
         lib.cademamix32bit_grad_bf16,
     ),
-    }
+}
 
-    str2optimizer8bit = {
+str2optimizer8bit = {
     "adam": (
         lib.cadam_static_8bit_grad_32,
         lib.cadam_static_8bit_grad_16,
@@ -81,9 +80,9 @@ if lib and lib.compiled_with_cuda:
         lib.cmomentum_static_8bit_grad_32,
         lib.cmomentum_static_8bit_grad_16,
     ),
-    }
+}
 
-    str2optimizer8bit_blockwise = {
+str2optimizer8bit_blockwise = {
     "adam": (
         lib.cadam_8bit_blockwise_grad_fp32,
         lib.cadam_8bit_blockwise_grad_fp16,
@@ -114,7 +113,7 @@ if lib and lib.compiled_with_cuda:
         lib.cademamix_8bit_blockwise_grad_fp16,
         lib.cademamix_8bit_blockwise_grad_bf16,
     ),
-    }
+}
 
 
 class GlobalPageManager: