Fix imports

bitsandbytes/__init__.py

@@ -4,7 +4,7 @@
 # LICENSE file in the root directory of this source tree.
 
 from .nn import modules
-from cextension import COMPILED_WITH_CUDA
+from .cextension import COMPILED_WITH_CUDA
 
 if COMPILED_WITH_CUDA:
     from .optim import adam

bitsandbytes/functional.py

@@ -9,7 +9,7 @@ from typing import Tuple
 import torch
 from torch import Tensor
 
-from cextension import lib, COMPILED_WITH_CUDA
+from .cextension import lib, COMPILED_WITH_CUDA
 
 name2qmap = {}
 

bitsandbytes/optim/__init__.py

@@ -13,4 +13,5 @@ if COMPILED_WITH_CUDA:
     from .lamb import LAMB, LAMB8bit, LAMB32bit
     from .rmsprop import RMSprop, RMSprop8bit, RMSprop32bit
     from .adagrad import Adagrad, Adagrad8bit, Adagrad32bit
-    from .optimizer import GlobalOptimManager
+
+from .optimizer import GlobalOptimManager