Skip to content

Commit

Permalink
default W4A16 alias to use group_size=128 (#94) (#95)
Browse files Browse the repository at this point in the history
  • Loading branch information
bfineran authored Jun 20, 2024
1 parent b77b1e6 commit 42575fd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/compressed_tensors/quantization/quant_scheme.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def is_preset_scheme(name: str) -> bool:
weights=QuantizationArgs(), input_activations=QuantizationArgs(symmetric=True)
)

W4A16 = dict(weights=QuantizationArgs(num_bits=4, symmetric=True))
W4A16 = dict(weights=QuantizationArgs(num_bits=4, group_size=128))

PRESET_SCHEMES = {
"W8A8": W8A8,
Expand Down

0 comments on commit 42575fd

Please sign in to comment.