update

bitsandbytes-foundation · Dec 5, 2024 · 7ba303f · 7ba303f
1 parent 632262f
commit 7ba303f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bitsandbytes/backends/npu.py b/bitsandbytes/backends/npu.py
@@ -90,7 +90,7 @@ def quantize_4bit(
         A: torch.Tensor,
         absmax: Optional[torch.Tensor] = None,
         out: Optional[torch.Tensor] = None,
-        blocksize=64,
+        blocksize: Optional[int] = None,
         compress_statistics=False,
         quant_type: Literal["fp4", "nf4"] = "nf4",
         quant_storage=torch.uint8,