QuantState.to(): move code tensor with others to correct device, fixes #1527 (#1528)

Author: matthewdouglas

bitsandbytes/functional.py

@@ -821,6 +821,7 @@ def as_dict(self, packed=False):
 
     def to(self, device):
         # make sure the quantization state is on the right device
+        self.code = self.code.to(device)
         self.absmax = self.absmax.to(device)
         if self.nested:
             self.offset = self.offset.to(device)