update

intel-analytics · Jul 30, 2024 · 06a5e5b · 06a5e5b
1 parent cf47053
commit 06a5e5b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/llm_performance_tests.yml b/.github/workflows/llm_performance_tests.yml
@@ -145,7 +145,7 @@ jobs:
           python -m pip install --upgrade expecttest
           bash python/llm/test/run-llm-install-tests.sh
 
-      - name: Test on xpu(transformers==4.41.2)
+      - name: Test on xpu(transformers==4.41.2 config)
         shell: bash
         run: |
           date_for_test_version=$(date -d yesterday +%Y-%m-%d)
@@ -184,7 +184,7 @@ jobs:
           python run.py
           mv *.csv test_batch4
 
-      - name: Test on xpu(transformers==4.41.2)
+      - name: Test on xpu(transformers==4.41.2 437)
         shell: bash
         run: |
           source /opt/intel/oneapi/setvars.sh
@@ -216,7 +216,7 @@ jobs:
           python run.py
           mv *.csv test_batch4
       
-      - name: Test on xpu(transformers==4.41.2)
+      - name: Test on xpu(transformers==4.41.2 440)
         shell: bash
         run: |
           source /opt/intel/oneapi/setvars.sh

diff --git a/python/llm/dev/benchmark/all-in-one/run.py b/python/llm/dev/benchmark/all-in-one/run.py
@@ -1990,6 +1990,8 @@ def run_pipeline_parallel_gpu(repo_id,
             batch_list = conf["batch_size"]
         for batch_size in batch_list:
             for model in conf.repo_id:
+                print("------------------------------------------------")
+                print("model:", model)
                 in_out_pairs = conf['in_out_pairs'].copy()
                 if excludes:
                     for in_out in conf['in_out_pairs']: