Skip to content

Commit c098447

Browse files
committed
fix typo
Signed-off-by: Kyle Sayers <[email protected]>
1 parent a6b3842 commit c098447

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llmcompressor/observers/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def get_qparams(
171171
group_indices, group_sizes = torch.unique(g_idx, return_counts=True)
172172
group_sizes = group_sizes[torch.argsort(group_indices)]
173173

174-
observed = observed.index_select(g_idx, -1)
174+
observed = observed.index_select(-1, g_idx)
175175

176176
# TODO: experiment with vectorizing for loop for performance
177177
end = 0

0 commit comments

Comments
 (0)