File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change 5
5
from keras .src import ops
6
6
from keras .src import testing
7
7
from keras .src .quantizers .gptq import GPTQ
8
- from keras .src .quantizers .gptqquant import GPTQQuant
8
+ from keras .src .quantizers .gptq_quant import GPTQQuant
9
9
10
10
11
11
def _get_mock_layer (layer_type , kernel_shape , rng ):
@@ -69,9 +69,7 @@ def test_full_quantization_process(self):
69
69
original_weights = np .copy (ops .convert_to_numpy (mock_layer .kernel ))
70
70
71
71
gptq_instance = GPTQ (mock_layer )
72
- gptq_instance .quantizer = GPTQQuant ()
73
- gptq_instance .quantizer .configure (wbits = 4 , symmetric = False )
74
-
72
+ gptq_instance .quantizer = GPTQQuant (wbits = 4 , symmetric = False )
75
73
calibration_data = rng .standard_normal (size = (128 , 16 )).astype ("float32" )
76
74
gptq_instance .update_hessian_with_batch (calibration_data )
77
75
gptq_instance .quantize_and_correct_block ()
You can’t perform that action at this time.
0 commit comments