Fix imports

bitsandbytes-foundation · Jul 1, 2022 · e4cf33f · e4cf33f
1 parent 8258b43
commit e4cf33f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/bitsandbytes/__init__.py b/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

diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
@@ -9,7 +9,7 @@
 import torch
 from torch import Tensor
 
-from cextension import lib, COMPILED_WITH_CUDA
+from .cextension import lib, COMPILED_WITH_CUDA
 
 name2qmap = {}
 

diff --git a/bitsandbytes/optim/__init__.py b/bitsandbytes/optim/__init__.py
@@ -13,4 +13,5 @@
     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