Skip to content

Commit 6e62e21

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 90327af commit 6e62e21

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

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

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -184,15 +184,15 @@ def _replace_param(
184184
if param.device.type == "meta":
185185
if isinstance(param, bnb.nn.Params4bit):
186186
return bnb.nn.Params4bit(
187-
data = data,
188-
requires_grad = data.requires_grad,
189-
quant_state = quant_state,
190-
blocksize = param.blocksize,
191-
compress_statistics = param.compress_statistics,
192-
quant_type = param.quant_type,
193-
quant_storage = param.quant_storage,
194-
module = param.module,
195-
bnb_quantized = param.bnb_quantized
187+
data=data,
188+
requires_grad=data.requires_grad,
189+
quant_state=quant_state,
190+
blocksize=param.blocksize,
191+
compress_statistics=param.compress_statistics,
192+
quant_type=param.quant_type,
193+
quant_storage=param.quant_storage,
194+
module=param.module,
195+
bnb_quantized=param.bnb_quantized,
196196
)
197197
return torch.nn.Parameter(data, requires_grad=data.requires_grad)
198198
param.data = data

0 commit comments

Comments
 (0)