diff --git a/python/llm/test/benchmark/igpu-perf/1024-128.yaml b/python/llm/test/benchmark/igpu-perf/1024-128.yaml index 98e05283..6d59984d 100644 --- a/python/llm/test/benchmark/igpu-perf/1024-128.yaml +++ b/python/llm/test/benchmark/igpu-perf/1024-128.yaml @@ -15,6 +15,7 @@ repo_id: - 'liuhaotian/llava-v1.5-7b' - 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-5-world-7b' + - 'IEITYuan/Yuan2-2B-hf' local_model_hub: 'path to your local model hub' warm_up: 1 num_trials: 3 diff --git a/python/llm/test/benchmark/igpu-perf/2048-256.yaml b/python/llm/test/benchmark/igpu-perf/2048-256.yaml index 598d2dd4..305f7864 100644 --- a/python/llm/test/benchmark/igpu-perf/2048-256.yaml +++ b/python/llm/test/benchmark/igpu-perf/2048-256.yaml @@ -14,6 +14,7 @@ repo_id: - 'liuhaotian/llava-v1.5-7b' - 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-5-world-7b' + - 'IEITYuan/Yuan2-2B-hf' local_model_hub: 'path to your local model hub' warm_up: 1 num_trials: 3 diff --git a/python/llm/test/benchmark/igpu-perf/32-32.yaml b/python/llm/test/benchmark/igpu-perf/32-32.yaml index e7c830b4..35f67366 100644 --- a/python/llm/test/benchmark/igpu-perf/32-32.yaml +++ b/python/llm/test/benchmark/igpu-perf/32-32.yaml @@ -15,6 +15,7 @@ repo_id: - 'liuhaotian/llava-v1.5-7b' - 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-5-world-7b' + - 'IEITYuan/Yuan2-2B-hf' local_model_hub: 'path to your local model hub' warm_up: 3 num_trials: 5 diff --git a/python/llm/test/benchmark/igpu-perf/32-512.yaml b/python/llm/test/benchmark/igpu-perf/32-512.yaml index 8f8060ef..d90b4082 100644 --- a/python/llm/test/benchmark/igpu-perf/32-512.yaml +++ b/python/llm/test/benchmark/igpu-perf/32-512.yaml @@ -15,6 +15,7 @@ repo_id: - 'liuhaotian/llava-v1.5-7b' - 'RWKV/rwkv-4-world-7b' - 'RWKV/rwkv-5-world-7b' + - 'IEITYuan/Yuan2-2B-hf' local_model_hub: 'path to your local model hub' warm_up: 1 num_trials: 3