Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit 1ed3862

Browse files
authored
update requirement for UT (#1058)
1 parent 5caf330 commit 1ed3862

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

.github/workflows/script/unitTest/run_unit_test_optimize.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ function pytest() {
2020
local coverage_log_dir=$1
2121
mkdir -p ${coverage_log_dir}
2222
pip install --no-cache-dir protobuf==3.20.0
23-
23+
## install transformers==4.34.1, to work with SharedDPO API
24+
pip install transformers==4.34.1
2425
cd /intel-extension-for-transformers/tests/CI || exit 1
2526
JOB_NAME=unit_test
2627
ut_log_name=${LOG_DIR}/${JOB_NAME}.log

conda_meta/meta.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
{% set version = "1.2" %}
1+
{% set version = "1.3" %}
22
{% set buildnumber = 0 %}
33
package:
44
name: intel_extension_for_transformers

tests/requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
datasets>=1.17
1+
datasets==2.14.7
22
mlflow
33
nlpaug==1.1.9
44
sacremoses
@@ -7,7 +7,7 @@ onnx>=1.10
77
onnxruntime
88
--find-links https://download.pytorch.org/cpu/whl/torch_stable.html
99
torch==2.1.0
10-
transformers==4.34.1
10+
transformers
1111
intel-tensorflow==2.13.0
1212
torchprofile
1313
intel-extension-for-pytorch

0 commit comments

Comments
 (0)