From 656de8ed110fce4e94b4f9d48494ecc5f8e04970 Mon Sep 17 00:00:00 2001 From: dbaranchuk Date: Tue, 23 Aug 2022 23:53:43 +0300 Subject: minor fixes --- bitsandbytes/autograd/_functions.py | 2 +- bitsandbytes/nn/modules.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'bitsandbytes') diff --git a/bitsandbytes/autograd/_functions.py b/bitsandbytes/autograd/_functions.py index 8ce1e60..641a779 100644 --- a/bitsandbytes/autograd/_functions.py +++ b/bitsandbytes/autograd/_functions.py @@ -368,7 +368,7 @@ class MatMul8bitLt(torch.autograd.Function): Bt = (CB * SCB).t().contiguous() CBt = (Bt / SCBt).t().to(torch.int8) - # intentionally, do not store CxBt into state + # intentionally, do not store CxBt in state CxBt, SBt = F.transform( CBt, to_order=formatB, transpose=True ) diff --git a/bitsandbytes/nn/modules.py b/bitsandbytes/nn/modules.py index ef7fefc..360a182 100644 --- a/bitsandbytes/nn/modules.py +++ b/bitsandbytes/nn/modules.py @@ -212,7 +212,7 @@ class Int8Params(torch.nn.Parameter): ) new_param.CB = self.CB new_param.SCB = self.SCB - new_param.SCB = self.SCBt + new_param.SCBt = self.SCBt return new_param -- cgit v1.2.3