pass current bnb_quantized when moving quantized Params4bit to different device (#1665)

mklabunde · matthewdouglas · commit bbb257dd2770 · 2025-07-01T16:22:23.000-04:00
(cherry picked from commit cd8bd2d)
diff --git a/bitsandbytes/nn/modules.py b/bitsandbytes/nn/modules.py
@@ -353,6 +353,7 @@ def to(self, *args, **kwargs):
                 compress_statistics=self.compress_statistics,
                 quant_type=self.quant_type,
                 quant_storage=self.quant_storage,
+                bnb_quantized=self.bnb_quantized,
             )
 
             return new_param

Original file line number	Diff line number	Diff line change
`@@ -353,6 +353,7 @@ def to(self, args, *kwargs):`
`353`	`353`	`compress_statistics=self.compress_statistics,`
`354`	`354`	`quant_type=self.quant_type,`
`355`	`355`	`quant_storage=self.quant_storage,`
	`356`	`+ bnb_quantized=self.bnb_quantized,`
`356`	`357`	`)`
`357`	`358`
`358`	`359`	`return new_param`