Add phi-2 to 2048-256 test for fixes (#10867)

2024-04-24 10:00:25 +08:00 · 2024-04-24 10:00:25 +08:00 · fb2a160af3
commit fb2a160af3
parent fabf54e052
1 changed files with 1 additions and 0 deletions
--- a/python/llm/test/benchmark/igpu-perf/2048-256_437.yaml
+++ b/python/llm/test/benchmark/igpu-perf/2048-256_437.yaml
@ -2,6 +2,7 @@ repo_id:
  - 'Qwen/Qwen1.5-7B-Chat'
  - 'meta-llama/Meta-Llama-3-8B-Instruct'
  - '01-ai/Yi-6B-Chat'
+  - 'microsoft/phi-2'
 local_model_hub: 'path to your local model hub'
 warm_up: 1
 num_trials: 3