We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9a1aba1 commit b8e3716Copy full SHA for b8e3716
src/compressed_tensors/compressors/base.py
@@ -200,7 +200,9 @@ def decompress_module(self, module: Module):
200
# Delete the old parameter and register the updated one
201
delete_offload_parameter(module, param_name)
202
offload_device = get_offloaded_device(module)
203
- param = torch.nn.Parameter(compressed_data[param_name], requires_grad=False)
+ param = torch.nn.Parameter(
204
+ compressed_data[param_name], requires_grad=False
205
+ )
206
register_offload_parameter(module, param_name, param, offload_device)
207
208
return result
0 commit comments