summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--bitsandbytes/__init__.py2
-rw-r--r--bitsandbytes/functional.py2
-rw-r--r--bitsandbytes/optim/__init__.py3
3 files changed, 4 insertions, 3 deletions
diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py
index 22fb841..02ca804 100644
--- 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
index 68b1d78..ab4e565 100644
--- a/bitsandbytes/functional.py
+++ b/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 = {}
diff --git a/bitsandbytes/optim/__init__.py b/bitsandbytes/optim/__init__.py
index e833ecc..42b5bc0 100644
--- a/bitsandbytes/optim/__init__.py
+++ b/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