From cad5c2f516646ffaaa5fb1d3b3928261214e060e Mon Sep 17 00:00:00 2001
From: "Chen, Zhentao" <zhentao.chen@intel.com>
Date: Mon, 8 Jan 2024 15:22:42 +0800
Subject: [PATCH] fixed harness deps version (#9854)

* fixed harness deps version

* fix typo
---
 .github/workflows/llm-harness-evaluation.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/llm-harness-evaluation.yml b/.github/workflows/llm-harness-evaluation.yml
index 77c68837..9bf292c4 100644
--- a/.github/workflows/llm-harness-evaluation.yml
+++ b/.github/workflows/llm-harness-evaluation.yml
@@ -132,7 +132,7 @@ jobs:
       - name: Run LLM install (all) test
         uses: ./.github/actions/llm/setup-llm-env
         with:
-          extra-dependency: "xpu"
+          extra-dependency: "xpu_2.0"
 
       - name: Install harness
         working-directory: ${{ github.workspace }}/python/llm/dev/benchmark/harness/
@@ -154,7 +154,7 @@ jobs:
       - name: Upgrade packages
         shell: bash
         run: |
-          pip install --upgrade transformers==4.34.0
+          pip install --upgrade transformers==4.34.0 datasets==2.14.6
 
       - name: Run harness
         shell: bash