fix typos, update overview img (#166)

Signed-off-by: Zhang, Weiwei1 <[email protected]>
intel · Jul 1, 2024 · 8d5765a · 8d5765a
1 parent f9e7d79
commit 8d5765a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/auto_round/autoround.py b/auto_round/autoround.py
@@ -338,11 +338,11 @@ def set_layerwise_config(self, weight_config):
         Returns:
         None
         """
-        layers_inblocks = get_layer_names_in_block(self.model, self.supported_types)
+        layers_in_blocks = get_layer_names_in_block(self.model, self.supported_types)
         for n, m in self.model.named_modules():
             if not isinstance(m, tuple(self.supported_types)):
                 continue
-            if n not in weight_config.keys() and n in layers_inblocks:
+            if n not in weight_config.keys() and n in layers_in_blocks:
                 weight_config[n] = {}
                 weight_config[n]["data_type"] = self.data_type
                 weight_config[n]["bits"] = self.bits

diff --git a/docs/imgs/autoround_overview.png b/docs/imgs/autoround_overview.png