[llm_bench] add support granite and granitemoe models (openvinotoolki…

…t#1486) related to huggingface/optimum-intel#1099 added opportunity to test these models via llm_bench Co-authored-by: Ilya Lavrenov <[email protected]>
xufang-lisa · Jan 7, 2025 · 09a5426 · 09a5426
1 parent 48dfd16
commit 09a5426
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'],
 }