Update test_activation_quantization_holder_gptq.py

sony · Dec 5, 2024 · 83c22f1 · 83c22f1
1 parent d3a34fb
commit 83c22f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/keras_tests/function_tests/test_activation_quantization_holder_gptq.py b/tests/keras_tests/function_tests/test_activation_quantization_holder_gptq.py
@@ -82,7 +82,7 @@ def _get_gptq_model(self, input_shape, get_model_fn):
                                graph=graph)
         trainer = KerasGPTQTrainer(graph,
                                    graph,
-                                   mct.gptq.get_keras_gptq_config(1, use_hessian_based_weights=False),
+                                   mct.gptq.get_keras_gptq_config(1, use_hessian_based_weights=False, use_hessian_sample_attention=False),
                                    keras_impl,
                                    DEFAULT_KERAS_INFO,
                                    representative_dataset)