From 305904cd8781b66efd8fa3bbddd9ce08eb9d6d55 Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Wed, 27 Nov 2024 11:26:07 +0000 Subject: [PATCH] Bugfix a typo --- src/compressed_tensors/compressors/sparse_compressors/base.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/compressed_tensors/compressors/sparse_compressors/base.py b/src/compressed_tensors/compressors/sparse_compressors/base.py index 67e2727a..d15057ce 100644 --- a/src/compressed_tensors/compressors/sparse_compressors/base.py +++ b/src/compressed_tensors/compressors/sparse_compressors/base.py @@ -119,8 +119,7 @@ def decompress( with safe_open(safe_path, framework="pt", device=device) as f: weight_data[param_name] = f.get_tensor(full_name) decompressed = self.decompress_weight(weight_data) - full_name = merge_names(weight_name, "weight") - yield full_name, decompressed + yield weight_name, decompressed for other_name, safe_path in other_params.items(): with safe_open(safe_path, framework="pt", device=device) as f: