Fix comment

44b3187a · Casper Hansen · 0f39807b · 44b3187a
Commit 44b3187a authored Sep 20, 2023 by Casper Hansen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

awq/quantize/quantizer.py awq/quantize/quantizer.py +1 -1

No files found.
--- a/awq/quantize/quantizer.py
+++ b/awq/quantize/quantizer.py
@@ -75,7 +75,7 @@ class AwqQuantizer:

            # [STEP 4]: Quantize weights
            for name, linear_layer in named_linears.items():
-                # NOTE: small regression in perplexity if linear layer you use .cpu().float()
+                # NOTE: small regression in perplexity if linear layer uses .cpu().float()
                linear_layer = linear_layer.cuda().half()

                linear_layer.weight.data, scales, zeros = self.pseudo_quantize_tensor(