[llm_bench] add support granite and granitemoe models

openvinotoolkit · Jan 6, 2025 · ea2dc52 · ea2dc52
1 parent b04b28b
commit ea2dc52
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'],
 }