[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/diffusion/compressor.py

@@ -29,6 +29,7 @@
 from auto_round.utils import (
     LazyImport,
     clear_memory,
+    config_save_pretrained,
     copy_python_files_from_model_cache,
     diffusion_load_model,
     dispatch_model_by_all_available_devices,
@@ -37,8 +38,6 @@
     get_block_names,
     merge_block_output_keys,
     wrap_block_forward_positional_to_kwargs,
-    dispatch_model_by_all_available_devices,
-    config_save_pretrained,
 )
 
 pipeline_utils = LazyImport("diffusers.pipelines.pipeline_utils")
@@ -519,9 +518,9 @@ def save_quantized(self, output_dir=None, format="auto_round", inplace=True, **k
         if not hasattr(self.pipe.config, "save_pretrained"):
             # For Z-Image, tuning will cause this attribute missing, we need to add it back before saving config
             setattr(
-                self.pipe.config, 
-                "save_pretrained", 
-                partial(config_save_pretrained, self.pipe.config, "model_index.json")
+                self.pipe.config,
+                "save_pretrained",
+                partial(config_save_pretrained, self.pipe.config, "model_index.json"),
             )
         self.pipe.config.save_pretrained(output_dir)
         return compressed_model

auto_round/utils/model.py

@@ -1956,6 +1956,7 @@ def _nextstep_pipeline_fn(pipe, prompts, guidance_scale=7.5, num_inference_steps
     pipe._autoround_pipeline_fn = _nextstep_pipeline_fn
     return pipe, model
 
+
 def config_save_pretrained(config, file_name, save_directory):
     if os.path.isfile(save_directory):
         raise AssertionError(f"Provided path ({save_directory}) should be a directory, not a file")