Skip to content

Commit b8e3716

Browse files
committed
quality
Signed-off-by: shanjiaz <zsjwpianpian@gmail.com>
1 parent 9a1aba1 commit b8e3716

File tree

1 file changed

+3
-1
lines changed
  • src/compressed_tensors/compressors

1 file changed

+3
-1
lines changed

src/compressed_tensors/compressors/base.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,9 @@ def decompress_module(self, module: Module):
200200
# Delete the old parameter and register the updated one
201201
delete_offload_parameter(module, param_name)
202202
offload_device = get_offloaded_device(module)
203-
param = torch.nn.Parameter(compressed_data[param_name], requires_grad=False)
203+
param = torch.nn.Parameter(
204+
compressed_data[param_name], requires_grad=False
205+
)
204206
register_offload_parameter(module, param_name, param, offload_device)
205207

206208
return result

0 commit comments

Comments
 (0)