summaryrefslogtreecommitdiff
path: root/bitsandbytes
diff options
context:
space:
mode:
authordbaranchuk <dmitrybaranchuk@gmail.com>2022-09-11 06:28:17 +0300
committerdbaranchuk <dmitrybaranchuk@gmail.com>2022-09-11 06:28:17 +0300
commit4dd475ced4adcbb31f6e1c42225f6d9b1e3be9f2 (patch)
treee005d5331249d2a0e1728a22eaf2a54528eb07e1 /bitsandbytes
parentd358999e9e2d98a834aaa38ffec1bef983d73fe6 (diff)
refactoring
Diffstat (limited to 'bitsandbytes')
-rw-r--r--bitsandbytes/autograd/_functions.py1
-rw-r--r--bitsandbytes/nn/modules.py2
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