Update optimum/exporters/openvino/model_configs.py

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>

Author: eaidova

optimum/exporters/openvino/model_configs.py

@@ -460,10 +460,8 @@ def __init__(
             random_sequence_length_range=random_sequence_length_range,
             **kwargs,
         )
-        if normalized_config.new_decoder_architecture and normalized_config.multi_query:
-            self.num_kv_heads = normalized_config.num_attention_heads
-        elif normalized_config.new_decoder_architecture and not normalized_config.multi_query:
-            self.num_kv_heads = normalized_config.num_kv_heads
+        if normalized_config.new_decoder_architecture:
+            self.num_kv_heads = normalized_config.num_attention_heads if normalized_config.multi_query else normalized_config.num_kv_heads
         else:
             self.num_kv_heads = 1