Skip to content

Commit

Permalink
groupsize => group_size
Browse files Browse the repository at this point in the history
  • Loading branch information
Michael Gschwind committed Apr 10, 2024
1 parent 0a6383a commit 43159c0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion quantize.py
Original file line number Diff line number Diff line change
Expand Up @@ -513,7 +513,7 @@ def __init__(self, mod, group_size=128, inner_k_tiles=8, padding_allowed=True):
self.groupsize = group_size
self.inner_k_tiles = inner_k_tiles
self.padding_allowed = padding_allowed
assert groupsize in [32, 64, 128, 256]
assert group_size in [32, 64, 128, 256]
assert inner_k_tiles in [2, 4, 8]

@torch.no_grad()
Expand Down

0 comments on commit 43159c0

Please sign in to comment.