Skip to content

Commit 89ef107

Browse files
fixed failing test case
1 parent baa0682 commit 89ef107

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

keras/src/quantizers/gptq_test.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from keras.src import ops
66
from keras.src import testing
77
from keras.src.quantizers.gptq import GPTQ
8-
from keras.src.quantizers.gptqquant import GPTQQuant
8+
from keras.src.quantizers.gptq_quant import GPTQQuant
99

1010

1111
def _get_mock_layer(layer_type, kernel_shape, rng):
@@ -69,9 +69,7 @@ def test_full_quantization_process(self):
6969
original_weights = np.copy(ops.convert_to_numpy(mock_layer.kernel))
7070

7171
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)
7573
calibration_data = rng.standard_normal(size=(128, 16)).astype("float32")
7674
gptq_instance.update_hessian_with_batch(calibration_data)
7775
gptq_instance.quantize_and_correct_block()

0 commit comments

Comments
 (0)