diff options
Diffstat (limited to 'bitsandbytes/__init__.py')
| -rw-r--r-- | bitsandbytes/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py index 6e5b6ac..76a5b48 100644 --- a/bitsandbytes/__init__.py +++ b/bitsandbytes/__init__.py @@ -12,6 +12,7 @@ from .autograd._functions import ( ) from .cextension import COMPILED_WITH_CUDA from .nn import modules +from . import cuda_setup if COMPILED_WITH_CUDA: from .optim import adam |