Skip to content

Commit

Permalink
Remove unnecesarry style change
Browse files Browse the repository at this point in the history
  • Loading branch information
rahul-tuli committed Jan 10, 2025
1 parent 26b23f7 commit cc4f78e
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit cc4f78e

Please sign in to comment.