Handle num_key_value_heads not present in config

openvinotoolkit · Dec 10, 2024 · cb2b3f6 · cb2b3f6
1 parent 40497d9
commit cb2b3f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/cpp/src/llm_pipeline_static.cpp b/src/cpp/src/llm_pipeline_static.cpp
@@ -413,7 +413,8 @@ ModelDesc get_modeldesc_from_json(const std::filesystem::path& filepath) {
     if (config_data.contains("_name_or_path")) {
         desc.name_or_path = config_data["_name_or_path"].get<std::string>();
     }
-    desc.num_key_value_heads = config_data["num_key_value_heads"].get<int>();
+    desc.num_key_value_heads = config_data.contains("num_key_value_heads")
+        ? config_data["num_key_value_heads"].get<int>() : -1;
     return desc;
 }