diff options
Diffstat (limited to 'bitsandbytes/__init__.py')
| -rw-r--r-- | bitsandbytes/__init__.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/bitsandbytes/__init__.py b/bitsandbytes/__init__.py index 7ca017d..6e5b6ac 100644 --- a/bitsandbytes/__init__.py +++ b/bitsandbytes/__init__.py @@ -3,8 +3,13 @@ # This source code is licensed under the MIT license found in the # LICENSE file in the root directory of this source tree. -from .autograd._functions import (MatmulLtState, bmm_cublas, matmul, - matmul_cublas, mm_cublas) +from .autograd._functions import ( + MatmulLtState, + bmm_cublas, + matmul, + matmul_cublas, + mm_cublas, +) from .cextension import COMPILED_WITH_CUDA from .nn import modules |