Skip to content

Commit aa9e75d

Browse files
committed
update supported model_ids
1 parent 34400b7 commit aa9e75d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tools/llm_bench/llm_bench_utils/config_class.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@
5353

5454
USE_CASES = {
5555
'image_gen': ['stable-diffusion-', 'ssd-', 'tiny-sd', 'small-sd', 'lcm-', 'sdxl', 'dreamlike', "flux"],
56-
"vlm": ["llava", "llava-next", "qwen2-vl", "llava-qwen2", "internvl-chat", "minicpmv", "phi3-v", "minicpm-v"],
56+
"vlm": ["llava", "llava-next", "qwen2-vl", "llava-qwen2", "internvl-chat", "minicpmv", "phi3-v", "minicpm-v", "maira2", "qwen2-5-vl"],
5757
'speech2text': ['whisper'],
5858
'image_cls': ['vit'],
5959
'code_gen': ['replit', 'codegen2', 'codegen', 'codet5', "stable-code"],
@@ -62,6 +62,7 @@
6262
't5',
6363
'falcon',
6464
"glm",
65+
"gpt",
6566
'gpt-',
6667
'gpt2',
6768
'aquila',
@@ -72,6 +73,7 @@
7273
'llama',
7374
'tiny-llama',
7475
'tinyllama',
76+
"opt",
7577
'opt-',
7678
'pythia-',
7779
'stablelm-',
@@ -93,6 +95,7 @@
9395
'mistral',
9496
'mixtral',
9597
'yi-',
98+
"phi",
9699
'phi-',
97100
'phi2-',
98101
'minicpm',

0 commit comments

Comments
 (0)