Skip to content

Commit 874111f

Browse files
Fix regression in 8bit parameter device movement (#1776)
1 parent f48050f commit 874111f

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

bitsandbytes/nn/modules.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -694,11 +694,10 @@ def to(self, *args, **kwargs):
694694
)
695695

696696
# If we had already quantized, move the statistics appropriately.
697-
if is_quantized and device is not None:
698-
if self.CB is not None:
699-
new_param.CB = new_param.data
697+
if is_quantized:
698+
new_param.CB = new_param.data
700699

701-
if self.SCB is not None:
700+
if self.SCB is not None and device is not None:
702701
new_param.SCB = self.SCB.to(device)
703702

704703
return new_param

0 commit comments

Comments
 (0)