Skip to content

Commit a217cb2

Browse files
committed
Merge remote-tracking branch 'origin' into kylesayrs/reduce-quantized-compression-memory
2 parents 049386c + 59f02b5 commit a217cb2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/compressed_tensors/compressors/model_compressors/model_compressor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -680,8 +680,8 @@ def _replace_weights(self, dense_weight_generator, model: Module):
680680
:param model: The model whose weights are to be updated.
681681
"""
682682

683-
for name, data in tqdm(dense_weight_generator, desc="Decompressing model"):
684-
module = operator.attrgetter(name)(model)
683+
for mod_path, data in tqdm(dense_weight_generator, desc="Decompressing model"):
684+
module = operator.attrgetter(mod_path)(model)
685685

686686
params_device = next(module.parameters()).device
687687
device = "cpu" if has_offloaded_params(module) else params_device

0 commit comments

Comments
 (0)