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

intel-analytics · Apr 24, 2024 · fb2a160 · fb2a160
1 parent fabf54e
commit fb2a160
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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