Skip to content

Commit 88e802c

Browse files
Merge pull request #1900 from TargetPackage/fix/params4bit-hf-initialized-kwarg
fix: accept **kwargs in Params4bit and Int8Params constructors
2 parents ecf9ca1 + ee7a536 commit 88e802c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bitsandbytes/nn/modules.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,7 @@ def __new__(
222222
quant_storage: torch.dtype = torch.uint8,
223223
module: Optional["Linear4bit"] = None,
224224
bnb_quantized: bool = False,
225+
**kwargs,
225226
) -> "Params4bit":
226227
if data is None:
227228
data = torch.empty(0)
@@ -680,6 +681,7 @@ def __new__(
680681
has_fp16_weights=False,
681682
CB: Optional[torch.Tensor] = None,
682683
SCB: Optional[torch.Tensor] = None,
684+
**kwargs,
683685
):
684686
if data is None:
685687
data = torch.empty(0)

0 commit comments

Comments
 (0)