Skip to content

Commit 383e8bc

Browse files
Fix CCT Test (use 8bit for weight_bit_width of Linear)
1 parent ac3cd41 commit 383e8bc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Tests/TestCCT.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def prepareCCT(model) -> nn.Module:
160160
"bias": False,
161161
"return_quant_tensor": True,
162162
"output_bit_width": 8,
163-
"weight_bit_width": 4,
163+
"weight_bit_width": 8,
164164
},
165165
),
166166
nn.Linear: (
@@ -172,7 +172,7 @@ def prepareCCT(model) -> nn.Module:
172172
"bias_quant": Int32Bias,
173173
"return_quant_tensor": True,
174174
"output_bit_width": 8,
175-
"weight_bit_width": 4,
175+
"weight_bit_width": 8,
176176
},
177177
),
178178
}

0 commit comments

Comments
 (0)