From 4d62f0c9c957545dfa78319cfe02948892637dca Mon Sep 17 00:00:00 2001
From: Nikita Savelyev <nikita.savelyev@intel.com>
Date: Wed, 6 Nov 2024 10:13:34 +0100
Subject: [PATCH] Add group size

---
 tests/openvino/test_quantization.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
index f44750510..2a15a2546 100644
--- a/tests/openvino/test_quantization.py
+++ b/tests/openvino/test_quantization.py
@@ -341,7 +341,7 @@ class OVWeightCompressionTest(unittest.TestCase):
                 True,
                 dict(
                     bits=4,
-                    group_size=-1,  # TODO: set to 32 after model is updated
+                    group_size=16,
                     dataset="contextual",
                     ratio=0.8,
                     sensitivity_metric="mean_activation_magnitude",
@@ -349,7 +349,7 @@ class OVWeightCompressionTest(unittest.TestCase):
                     processor=MODEL_NAMES["minicpmv"],
                     trust_remote_code=True,
                 ),
-                {"int4": 24, "int8": 6},
+                {"int4": 22, "int8": 8},
             )
         )