update fabric bitsandbytes linear quantization for bnb 0.44.1

Author: Ali Alshaarawy

src/lightning/fabric/plugins/precision/bitsandbytes.py

@@ -39,11 +39,11 @@
 
 log = logging.getLogger(__name__)
 
-_BITSANDBYTES_AVAILABLE = RequirementCache("bitsandbytes>=0.42.0")
+_BITSANDBYTES_AVAILABLE = RequirementCache("bitsandbytes>=0.44.0")
 
 
 class BitsandbytesPrecision(Precision):
-    """Plugin for quantizing weights with `bitsandbytes <https://github.com/TimDettmers/bitsandbytes>`__.
+    """Plugin for quantizing weights with `bitsandbytes <https://github.com/bitsandbytes-foundation/bitsandbytes>`__.
 
     .. warning::  This is an :ref:`experimental <versioning:Experimental API>` feature.
 
@@ -322,6 +322,7 @@ def quantize_(self, weight: Optional[torch.Tensor] = None, device: Optional[torc
                 return
             assert isinstance(self.weight, bnb.nn.Params4bit)
             self.weight = self.quantize(self.weight, weight, device)
+            self.weight.bnb_quantized = True
 
         @staticmethod
         def quantize(

src/lightning/pytorch/plugins/precision/bitsandbytes.py

@@ -16,7 +16,7 @@
 
 
 class BitsandbytesPrecision(Precision, FabricBNBPrecision):
-    """Plugin for quantizing weights with `bitsandbytes <https://github.com/TimDettmers/bitsandbytes>`__.
+    """Plugin for quantizing weights with `bitsandbytes <https://github.com/bitsandbytes-foundation/bitsandbytes>`__.
 
     .. warning::  This is an :ref:`experimental <versioning:Experimental API>` feature.