Merge branch 'kaihui/ar_new_api' of https://github.com/intel/neural-c…

…ompressor into kaihui/ar_new_api
intel · Apr 28, 2024 · 9b86550 · 9b86550
2 parents 16f82d1 + abf6d74
commit 9b86550
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/neural_compressor/torch/algorithms/weight_only/autoround.py b/neural_compressor/torch/algorithms/weight_only/autoround.py
@@ -253,8 +253,7 @@ def get_autoround_default_run_fn(
 class AutoRoundProcessor(AutoRound):
 
     def prepare(self):
-        """Prepares a given model for quantization.
-        """
+        """Prepares a given model for quantization."""
         # logger.info("cache block input")
         self.start_time = time.time()
         self.block_names = get_block_names(self.model)
@@ -273,8 +272,7 @@ def prepare(self):
         self._replace_forward()
 
     def convert(self):
-        """Converts a prepared model to a quantized model.
-        """
+        """Converts a prepared model to a quantized model."""
         self._recover_forward()
         inputs = self.inputs[self.tmp_block_name]
         del self.tmp_block_name