Skip to content

Commit f325660

Browse files
committed
for cpu
1 parent 9e5518e commit f325660

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/llm_performance_tests.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -419,6 +419,8 @@ jobs:
419419
export https_proxy=${HTTPS_PROXY}
420420
source ipex-llm-init -t
421421
export OMP_NUM_THREADS=48
422+
# upgrade for default transformers version
423+
python -m pip install transformers==4.41.2, trl
422424
# hide time info
423425
sed -i 's/str(end - st)/"xxxxxx"/g' run.py
424426
python run.py
@@ -499,6 +501,8 @@ jobs:
499501
cd python/llm/dev/benchmark/all-in-one
500502
export http_proxy=${HTTP_PROXY}
501503
export https_proxy=${HTTPS_PROXY}
504+
# upgrade for default transformers version
505+
python -m pip install transformers==4.41.2, trl
502506
# hide time info
503507
sed -i 's/str(end - st)/"xxxxxx"/g' run.py
504508
python run.py

0 commit comments

Comments
 (0)