We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a6b3842 commit c098447Copy full SHA for c098447
src/llmcompressor/observers/base.py
@@ -171,7 +171,7 @@ def get_qparams(
171
group_indices, group_sizes = torch.unique(g_idx, return_counts=True)
172
group_sizes = group_sizes[torch.argsort(group_indices)]
173
174
- observed = observed.index_select(g_idx, -1)
+ observed = observed.index_select(-1, g_idx)
175
176
# TODO: experiment with vectorizing for loop for performance
177
end = 0
0 commit comments