From ff2b26acf917d05e2e8b594495b09d2a13571c15 Mon Sep 17 00:00:00 2001 From: Rahul Tuli Date: Fri, 10 Jan 2025 14:09:45 +0000 Subject: [PATCH] Remove unnecesarry style change --- .../compressors/model_compressors/model_compressor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/compressed_tensors/compressors/model_compressors/model_compressor.py b/src/compressed_tensors/compressors/model_compressors/model_compressor.py index 78ab6422..4ad99241 100644 --- a/src/compressed_tensors/compressors/model_compressors/model_compressor.py +++ b/src/compressed_tensors/compressors/model_compressors/model_compressor.py @@ -269,9 +269,9 @@ def compress( compressed_state_dict = state_dict - quantized_modules_to_args: Dict[str, QuantizationArgs] = ( - map_modules_to_quant_args(model) - ) + quantized_modules_to_args: Dict[ + str, QuantizationArgs + ] = map_modules_to_quant_args(model) if self.quantization_compressor is not None: compressed_state_dict = self.quantization_compressor.compress(