From 88da7f7c1682740653ef489d722e1e8645cc173e Mon Sep 17 00:00:00 2001 From: dbogunowicz <97082108+dbogunowicz@users.noreply.github.com> Date: Fri, 17 May 2024 17:46:50 +0200 Subject: [PATCH] Update sparse_bitmask.py (#56) --- src/compressed_tensors/compressors/sparse_bitmask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/compressed_tensors/compressors/sparse_bitmask.py b/src/compressed_tensors/compressors/sparse_bitmask.py index 10b398c1..c0cc1888 100644 --- a/src/compressed_tensors/compressors/sparse_bitmask.py +++ b/src/compressed_tensors/compressors/sparse_bitmask.py @@ -67,7 +67,7 @@ def compress(self, model_state: Dict[str, Tensor]) -> Dict[str, Tensor]: f"found an existing entry for {key}. The existing entry will " "be replaced." ) - compressed_dict |= bitmask_dict + compressed_dict.update(bitmask_dict) return compressed_dict