Skip to content

Commit 3e7fb58

Browse files
[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>
1 parent 48dfd16 commit 3e7fb58

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tools/llm_bench/llm_bench_utils/config_class.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,9 @@
102102
"olmo",
103103
"phi3",
104104
"starcoder",
105-
"instruct-gpt"
105+
"instruct-gpt",
106+
"granite",
107+
"granitemoe",
106108
],
107109
'ldm_super_resolution': ['ldm-super-resolution'],
108110
}

0 commit comments

Comments
 (0)