diff options
| author | dbaranchuk <dmitrybaranchuk@gmail.com> | 2022-09-11 06:28:17 +0300 |
|---|---|---|
| committer | dbaranchuk <dmitrybaranchuk@gmail.com> | 2022-09-11 06:28:17 +0300 |
| commit | 4dd475ced4adcbb31f6e1c42225f6d9b1e3be9f2 (patch) | |
| tree | e005d5331249d2a0e1728a22eaf2a54528eb07e1 | |
| parent | d358999e9e2d98a834aaa38ffec1bef983d73fe6 (diff) | |
refactoring
| -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 |