Bugfix a typo

neuralmagic · Nov 27, 2024 · 305904c · 305904c
1 parent a528334
commit 305904c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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: