Skip to content

Commit 5bf957f

Browse files
revert extraneous test change
Signed-off-by: Brian Dellabetta <[email protected]>
1 parent 76f81b9 commit 5bf957f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_quantization/lifecycle/test_apply.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ def test_target_prioritization(mock_frozen):
6666
"config_groups": {
6767
"group_1": {
6868
"weights": {
69-
"num_bits": 6,
69+
"num_bits": 8,
7070
},
7171
"targets": ["Linear"],
7272
},
@@ -101,7 +101,7 @@ def test_target_prioritization(mock_frozen):
101101
elif re.match(".*down_proj", name):
102102
assert module.quantization_scheme.weights.num_bits == 4
103103
elif isinstance(module, torch.nn.Linear):
104-
assert module.quantization_scheme.weights.num_bits == 6
104+
assert module.quantization_scheme.weights.num_bits == 8
105105

106106

107107
def test_apply_quantization_config_tinyllama():

0 commit comments

Comments
 (0)