From 4dd475ced4adcbb31f6e1c42225f6d9b1e3be9f2 Mon Sep 17 00:00:00 2001 From: dbaranchuk Date: Sun, 11 Sep 2022 06:28:17 +0300 Subject: refactoring --- bitsandbytes/autograd/_functions.py | 1 - bitsandbytes/nn/modules.py | 2 +- 2 files changed, 1 insertion(+), 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 -- cgit v1.2.3