File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 53
53
pip install transformers==4.2.0
54
54
pip uninstall -y h5py
55
55
pip install h5py==2.9.0
56
- pip install numpy==1.19.0
56
+ pip install numpy==1.19.0 tensorflow==${{ inputs.tf_version }}
57
57
else
58
58
pip install transformers
59
59
if [[ "${{ inputs.tf_version }}" != "2.13.0" && "${{ inputs.tf_version }}" != "2.9.0" ]]; then
Original file line number Diff line number Diff line change 32
32
pip install onnxruntime==${{ inputs.ort_version }}
33
33
pip uninstall -y protobuf
34
34
if [[ ${{ inputs.tf_version }} == 1.* ]]; then
35
- pip install numpy==1.19.0
35
+ pip install numpy==1.19.0 tensorflow==${{ inputs.tf_version }}
36
36
else
37
37
if [[ "${{ inputs.tf_version }}" != "2.13.0" && "${{ inputs.tf_version }}" != "2.9.0" ]]; then
38
38
echo "-- install-2 TF-KERAS ${{ inputs.tf_version }}"
Original file line number Diff line number Diff line change @@ -20,7 +20,7 @@ pip uninstall -y tensorflow
20
20
pip install onnx==$ONNX_VERSION onnxruntime==$ORT_VERSION onnxruntime-extensions
21
21
22
22
if [[ $TF_VERSION == 1.* ]]; then
23
- pip install numpy==1.19.0
23
+ pip install numpy==1.19.0 tensorflow== $TF_VERSION
24
24
else
25
25
pip uninstall -y protobuf
26
26
if [[ " $TF_VERSION " != " 2.13.0" && " $TF_VERSION " != " 2.9.0" ]]; then
You can’t perform that action at this time.
0 commit comments