diff --git a/src/llmcompressor/modifiers/awq/base.py b/src/llmcompressor/modifiers/awq/base.py index 5c187c17e..7d45820de 100644 --- a/src/llmcompressor/modifiers/awq/base.py +++ b/src/llmcompressor/modifiers/awq/base.py @@ -397,7 +397,7 @@ def cache_smooth_activations_hook( ): self._smooth_activation_means[smooth_name] = _accumulate_mean( # Assume that first argument is the input - args[0].cpu().detach().squeeze(), + args[0].cpu().abs().detach().squeeze(), self._smooth_activation_means.get(smooth_name, None), )