diff options
Diffstat (limited to 'bitsandbytes/__init__.py')
| -rw-r--r-- | bitsandbytes/__init__.py | 2 |
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 |