Skip to content

Commit 78222ba

Browse files
committed
lint
Signed-off-by: George Ohashi <george@neuralmagic.com>
1 parent 189e9d5 commit 78222ba

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/llmcompressor/observers/base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,8 @@ def get_qparams(
142142
) # shape of [num_key_value_heads * head_dim]
143143
else:
144144
# weight or input
145-
# assume observed is transposed, because its the output, hence use dim 0
145+
# assume observed is transposed,
146+
# because its the output, hence use dim 0
146147
self._scale, self._zero_point = self.get_qparams_along_dim(
147148
observed, 0
148149
)

0 commit comments

Comments
 (0)