Skip to content

Commit ac3cd41

Browse files
Fix CCT Test Pretrained
1 parent 29d7c0c commit ac3cd41

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Tests/TestCCTPretrained.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def prepareFQCCT(model) -> nn.Module:
208208
"bias": False,
209209
"return_quant_tensor": True,
210210
"output_bit_width": 8,
211-
"weight_bit_width": 4,
211+
"weight_bit_width": 8,
212212
},
213213
),
214214
nn.Linear: (
@@ -220,7 +220,7 @@ def prepareFQCCT(model) -> nn.Module:
220220
"bias_quant": Int32Bias,
221221
"return_quant_tensor": True,
222222
"output_bit_width": 8,
223-
"weight_bit_width": 4,
223+
"weight_bit_width": 8,
224224
},
225225
),
226226
}

0 commit comments

Comments
 (0)