[llm_bench] add support granite and granitemoe models (#1486)

related to huggingface/optimum-intel#1099 added opportunity to test these models via llm_bench Co-authored-by: Ilya Lavrenov <ilya.lavrenov@intel.com>
openvinotoolkit · Jan 7, 2025 · 3e7fb58 · 3e7fb58
1 parent 48dfd16
commit 3e7fb58
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/llm_bench/llm_bench_utils/config_class.py b/tools/llm_bench/llm_bench_utils/config_class.py
@@ -102,7 +102,9 @@
         "olmo",
         "phi3",
         "starcoder",
-        "instruct-gpt"
+        "instruct-gpt",
+        "granite",
+        "granitemoe",
     ],
     'ldm_super_resolution': ['ldm-super-resolution'],
 }