Fix typo (#486)

casper-hansen · Jun 8, 2024 · 76ac8c7 · 76ac8c7
1 parent 2627364
commit 76ac8c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/awq/models/base.py b/awq/models/base.py
@@ -591,7 +591,7 @@ def _load_quantized_modules(
             # Get every linear layer in a block
             named_linears = get_named_linears(layer)
 
-            # Filter out the linear layers we don't want to exclude
+            # Filter out the linear layers we don't want to include
             named_linears = exclude_layers_to_not_quantize(
                 named_linears, quant_config.modules_to_not_convert
             )