diff options
| -rw-r--r-- | bitsandbytes/__init__.py | 2 | ||||
| -rw-r--r-- | bitsandbytes/functional.py | 2 | ||||
| -rw-r--r-- | bitsandbytes/optim/__init__.py | 3 |
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 |