[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

@@ -86,8 +86,8 @@
     is_mx_fp,
     is_nv_fp,
     is_standard_fp,
-    is_torch_compile_enabled,
     is_static_wfp8afp8,
+    is_torch_compile_enabled,
     is_wfp8afp8,
     llm_load_model,
     mv_module_from_gpu,
@@ -2095,9 +2095,9 @@ def _get_block_outputs(
             tmp_input_ids, tmp_input_others = self._sampling_inputs(
                 input_ids, input_others, indices, self.seqlen, self.batch_dim, share_cache_keys=self.shared_cache_keys
             )
-            tmp_output = self.block_forward(block, tmp_input_ids, tmp_input_others, self.amp, self.amp_dtype, device).to(
-                cache_device
-            )
+            tmp_output = self.block_forward(
+                block, tmp_input_ids, tmp_input_others, self.amp, self.amp_dtype, device
+            ).to(cache_device)
             if save_output:
                 if self.batch_size == 1:
                     output.append(tmp_output)

auto_round/compressors/diffusion/compressor.py

@@ -28,12 +28,12 @@
     LazyImport,
     block_forward,
     clear_memory,
+    compile_func,
     diffusion_load_model,
     extract_block_names_to_str,
     find_matching_blocks,
     get_block_names,
     is_torch_compile_enabled,
-    compile_func,
 )
 
 pipeline_utils = LazyImport("diffusers.pipelines.pipeline_utils")
@@ -208,7 +208,9 @@ def _get_current_q_output(
             hidden_states = current_input_ids.pop("hidden_states")
             current_input_others.update(current_input_ids)
             current_input_ids = hidden_states
-        output_q = self.block_forward(block, current_input_ids, current_input_others, self.amp, self.amp_dtype, device, idx)
+        output_q = self.block_forward(
+            block, current_input_ids, current_input_others, self.amp, self.amp_dtype, device, idx
+        )
         return output_q
 
     @torch.no_grad()
@@ -252,7 +254,9 @@ def _get_block_outputs(
                 tmp_input_others.update(tmp_input_ids)
                 tmp_input_ids = hidden_states
 
-            tmp_output = self.block_forward(block, tmp_input_ids, tmp_input_others, self.amp, self.amp_dtype, device, None)
+            tmp_output = self.block_forward(
+                block, tmp_input_ids, tmp_input_others, self.amp, self.amp_dtype, device, None
+            )
             assert len(output_config) == len(tmp_output)
             tmp_output = dict(zip(output_config, tmp_output))
 

auto_round/wrapper.py

@@ -22,13 +22,13 @@
 from .utils import (
     SUPPORTED_LAYER_TYPES,
     check_to_quantized,
+    compile_func,
     deepspeed_exists,
     get_scale_shape,
     is_mx_fp,
     is_nv_fp,
-    set_module,
     is_torch_compile_enabled,
-    compile_func,
+    set_module,
 )
 
 if deepspeed_exists: