From 96f376ea4278fb7bfdbb2832794b7809c141596d Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Wed, 22 Jan 2025 21:39:31 +0000 Subject: [PATCH] Lint Signed-off-by: Rahul Tuli --- .../compressed_tensors/schemes/compressed_tensors_24.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_24.py b/vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_24.py index 785577a4d4170..da652e00bde8d 100644 --- a/vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_24.py +++ b/vllm/model_executor/layers/quantization/compressed_tensors/schemes/compressed_tensors_24.py @@ -264,7 +264,8 @@ def _decompress_bitmask_compressed_weight( :param compressed: The 2:4 sparse weight tensor compressed using the sparse-24-bitmask compressor. :param bitmask: The 2:4 bitmask associated with the compressed weights. - :param layer: The layer whose weights need to be processed after loading. + :param layer: The layer whose weights need to be processed + after loading. :return: The decompressed 2:4 sparse weight tensor. """