Skip to content

Commit d492543

Browse files
committed
added TODO
Signed-off-by: shanjiaz <[email protected]>
1 parent 2bf6e19 commit d492543

File tree

1 file changed

+4
-0
lines changed
  • src/compressed_tensors/compressors

1 file changed

+4
-0
lines changed

src/compressed_tensors/compressors/base.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,10 @@ def decompress_module(self, module: Module):
190190
for name, parameter in module.named_parameters():
191191
compressed_data[name] = parameter
192192

193+
# NOTE: decompress_weight may modify compressed_data dict in-place
194+
# This is subtle but allows us to update the module's qparams with
195+
# the unpacked values.
196+
# TODO: Consider refactoring to return modified qparams explicitly
193197
result = self.decompress_weight(
194198
compressed_data=compressed_data, quantization_args=quantization_args
195199
).to(device)

0 commit comments

Comments
 (0)