diff options
| -rw-r--r-- | bitsandbytes/autograd/_functions.py | 1 | ||||
| -rw-r--r-- | bitsandbytes/nn/modules.py | 2 |
2 files changed, 1 insertions, 2 deletions
diff --git a/bitsandbytes/autograd/_functions.py b/bitsandbytes/autograd/_functions.py index 642e516..48f867f 100644 --- a/bitsandbytes/autograd/_functions.py +++ b/bitsandbytes/autograd/_functions.py @@ -1,6 +1,5 @@ import operator import torch -import bitsandbytes as bnb import bitsandbytes.functional as F from dataclasses import dataclass diff --git a/bitsandbytes/nn/modules.py b/bitsandbytes/nn/modules.py index 00d0c61..e7e759d 100644 --- a/bitsandbytes/nn/modules.py +++ b/bitsandbytes/nn/modules.py @@ -221,9 +221,9 @@ class Linear8bitLt(nn.Linear): output_features, bias=True, has_fp16_weights=True, + memory_efficient_backward=False, threshold=0.0, index=None, - memory_efficient_backward=False ): super(Linear8bitLt, self).__init__( input_features, output_features, bias |