summaryrefslogtreecommitdiff
path: root/bitsandbytes
diff options
context:
space:
mode:
authorMax Ryabinin <mryabinin0@gmail.com>2022-07-01 17:25:44 +0300
committerMax Ryabinin <mryabinin0@gmail.com>2022-07-01 17:25:44 +0300
commite4cf33f2a38e54ca2a523bc5d5923237c68b0112 (patch)
treec47ed406d5c465476078181c4889b7e7733e16e9 /bitsandbytes
parent8258b4364a21a4da2572cb644d0926080c3268da (diff)
Fix imports
Diffstat (limited to 'bitsandbytes')
-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