File tree Expand file tree Collapse file tree 3 files changed +11
-13
lines changed Expand file tree Collapse file tree 3 files changed +11
-13
lines changed Original file line number Diff line number Diff line change @@ -46,14 +46,11 @@ runs:
46
46
pip install git+https://github.com/qubvel/efficientnet
47
47
pip install keras-self-attention
48
48
pip install pytest pytest-cov pytest-runner
49
- pip uninstall -y protobuf
49
+ pip uninstall -y protobuf h5py
50
50
51
51
if [[ ${{ inputs.tf_version }} == 1.* ]]; then
52
- pip install keras==2.3.1
53
- pip install transformers==4.2.0
54
- pip uninstall -y h5py
55
- pip install h5py==2.9.0
56
- pip install numpy==1.19.0 tensorflow==${{ inputs.tf_version }}
52
+ echo "-- install-1 TF1-KERAS ${{ inputs.tf_version }}"
53
+ pip install numpy==1.19.0 tensorflow==${{ inputs.tf_version }} protobuf h5py==2.9.0 transformers==4.2.0 keras h5py
57
54
else
58
55
pip install transformers
59
56
if [[ "${{ inputs.tf_version }}" != "2.13.0" && "${{ inputs.tf_version }}" != "2.9.0" ]]; then
Original file line number Diff line number Diff line change @@ -24,22 +24,22 @@ runs:
24
24
python -m pip install --upgrade pip
25
25
pip install onnxconverter-common
26
26
pip install onnx==${{ inputs.onnx_version }}
27
- pip install h5py
28
27
pip install parameterized
29
28
pip install timeout-decorator
30
29
pip install coloredlogs flatbuffers
31
30
pip install pytest pytest-cov pytest-runner
32
31
pip install onnxruntime==${{ inputs.ort_version }}
33
- pip uninstall -y protobuf
32
+ pip uninstall -y protobuf h5py tensorflow
34
33
if [[ ${{ inputs.tf_version }} == 1.* ]]; then
35
- pip install numpy==1.19.0 tensorflow==${{ inputs.tf_version }}
34
+ echo "-- install-2 TF1-KERAS ${{ inputs.tf_version }}"
35
+ pip install numpy==1.19.0 tensorflow==${{ inputs.tf_version }} protobuf keras h5py
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 }}"
39
- pip install tf_keras==${{ inputs.tf_version }} tensorflow==${{ inputs.tf_version }}
39
+ pip install tf_keras==${{ inputs.tf_version }} tensorflow==${{ inputs.tf_version }} h5py
40
40
else
41
41
echo "-- install-2 TF ${{ inputs.tf_version }}"
42
- pip install protobuf tensorflow==${{ inputs.tf_version }}
42
+ pip install protobuf tensorflow==${{ inputs.tf_version }} h5py
43
43
fi
44
44
fi
45
45
Original file line number Diff line number Diff line change @@ -16,11 +16,12 @@ echo "==== ONNXRuntime version: $ORT_VERSION"
16
16
echo " ==== ONNX version: $ONNX_VERSION "
17
17
18
18
pip install pytest pytest-cov pytest-runner coverage graphviz requests pyyaml pillow pandas parameterized sympy coloredlogs flatbuffers timeout-decorator
19
- pip uninstall -y tensorflow
19
+ pip uninstall -y tensorflow protobuf h5py
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 tensorflow==$TF_VERSION
23
+ echo " -- install-3 TF1-KERAS $TF_VERSION "
24
+ pip install numpy==1.19.0 tensorflow==$TF_VERSION protobug keras h5py
24
25
else
25
26
pip uninstall -y protobuf
26
27
if [[ " $TF_VERSION " != " 2.13.0" && " $TF_VERSION " != " 2.9.0" ]]; then
You can’t perform that action at this time.
0 commit comments