[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

auto_round/compressors/base.py

@@ -66,6 +66,7 @@
     estimate_tuning_block_mem,
     find_matching_blocks,
     flatten_list,
+    get_avg_bits,
     get_block_names,
     get_device_memory,
     get_fp_layer_names,
@@ -95,7 +96,6 @@
     to_device,
     to_dtype,
     unsupport_meta_device,
-    get_avg_bits,
 )
 from auto_round.wrapper import WrapperLinear, WrapperMultiblock, unwrapper_block, unwrapper_layer, wrapper_block
 
@@ -1695,7 +1695,9 @@ def quantize(self) -> tuple[torch.nn.Module, dict[str, Any]]:
         average_bits_w_lm_head = get_avg_bits(self.model, with_lm_head=True)
         if average_bits_w_lm_head != average_bits:
             logger.info(f"The target average bits of blocks in the model (without lm_head): {average_bits:.3f} bits")
-            logger.info(f"The target average bits of the entire model (with lm_head): {average_bits_w_lm_head:.3f} bits")
+            logger.info(
+                f"The target average bits of the entire model (with lm_head): {average_bits_w_lm_head:.3f} bits"
+            )
         else:
             logger.info(f"The target average bits of the entire model: {average_bits:.3f} bits")
         if not hasattr(self, "formats"):

auto_round/utils.py

@@ -2769,7 +2769,7 @@ def get_avg_bits(module, with_lm_head=False):
                 if m.data_type in ("fp4_v2", "nv_fp"):
                     scale_bits += 32  # global scale bits
                 all_bits += scale_bits
-            else: # woq
+            else:  # woq
                 scale_bits = 16 * (m_numel // m.group_size)
                 all_bits += scale_bits