diff options
| author | dbaranchuk <dmitrybaranchuk@gmail.com> | 2022-08-23 23:53:43 +0300 |
|---|---|---|
| committer | dbaranchuk <dmitrybaranchuk@gmail.com> | 2022-08-23 23:53:43 +0300 |
| commit | 656de8ed110fce4e94b4f9d48494ecc5f8e04970 (patch) | |
| tree | 05a85afe65454fa04434e5f4fa82c1f370651e3b | |
| parent | 1753aa04185b10a3bb52f7289ed4af15cf2502a7 (diff) | |
minor fixes
| -rw-r--r-- | bitsandbytes/autograd/_functions.py | 2 | ||||
| -rw-r--r-- | bitsandbytes/nn/modules.py | 2 |
2 files changed, 2 insertions, 2 deletions
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 |