summaryrefslogtreecommitdiff
path: root/bitsandbytes/__init__.py
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/__init__.py
parent8258b4364a21a4da2572cb644d0926080c3268da (diff)
Fix imports
Diffstat (limited to 'bitsandbytes/__init__.py')
-rw-r--r--bitsandbytes/__init__.py2
1 files changed, 1 insertions, 1 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