diff options
| author | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:25:44 +0300 |
|---|---|---|
| committer | Max Ryabinin <mryabinin0@gmail.com> | 2022-07-01 17:25:44 +0300 |
| commit | e4cf33f2a38e54ca2a523bc5d5923237c68b0112 (patch) | |
| tree | c47ed406d5c465476078181c4889b7e7733e16e9 | |
| parent | 8258b4364a21a4da2572cb644d0926080c3268da (diff) | |
Fix imports
| -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 |