@@ -94,7 +94,8 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
94
94
COMMIT_TRANSFORMERS=$( python tools/yaml_utils.py -f dependency_version.yml -d transformers -k commit)
95
95
VER_PROTOBUF=$( python tools/yaml_utils.py -f dependency_version.yml -d protobuf -k version)
96
96
VER_LM_EVAL=$( python tools/yaml_utils.py -f dependency_version.yml -d lm_eval -k version)
97
- VER_NUMPY=$( python tools/yaml_utils.py -f dependency_version.yml -d numpy -k version)
97
+ VER_NUMPY_MIN=$( python tools/yaml_utils.py -f dependency_version.yml -d numpy -k version_min)
98
+ VER_NUMPY_MAX=$( python tools/yaml_utils.py -f dependency_version.yml -d numpy -k version_max)
98
99
VER_SETUPTOOLS=$( python tools/yaml_utils.py -f dependency_version.yml -d setuptools -k version)
99
100
VER_IPEX_MAJOR=$( grep " VERSION_MAJOR" version.txt | cut -d " " -f 2)
100
101
VER_IPEX_MINOR=$( grep " VERSION_MINOR" version.txt | cut -d " " -f 2)
@@ -116,7 +117,8 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
116
117
mkdir ${WHEELFOLDER}
117
118
118
119
# Install deps
119
- python -m pip install cmake ninja numpy==${VER_NUMPY} setuptools==${VER_SETUPTOOLS}
120
+ python -m pip install cmake ninja setuptools==${VER_SETUPTOOLS}
121
+ python -m pip install " numpy >= ${VER_NUMPY_MIN} ,< ${VER_NUMPY_MAX} "
120
122
121
123
echo " #!/bin/bash" > ${AUX_INSTALL_SCRIPT}
122
124
if [ $(( ${MODE} & 0x04 )) -ne 0 ]; then
@@ -146,7 +148,8 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
146
148
fi
147
149
148
150
# echo "python -m pip install impi-devel" >> ${AUX_INSTALL_SCRIPT}
149
- echo " python -m pip install numpy==${VER_NUMPY} setuptools==${VER_SETUPTOOLS} " >> ${AUX_INSTALL_SCRIPT}
151
+ echo " python -m pip install setuptools==${VER_SETUPTOOLS} " >> ${AUX_INSTALL_SCRIPT}
152
+ echo " python -m pip install \" numpy >=${VER_NUMPY_MIN} , <${VER_NUMPY_MAX} \" " >> ${AUX_INSTALL_SCRIPT}
150
153
echo " python -m pip install cpuid accelerate datasets sentencepiece diffusers mpi4py protobuf==${VER_PROTOBUF} lm_eval==${VER_LM_EVAL} huggingface_hub py-cpuinfo " >> ${AUX_INSTALL_SCRIPT}
151
154
echo " python -m pip install deepspeed==${VER_DS} " >> ${AUX_INSTALL_SCRIPT}
152
155
echo " python -m pip install bitsandbytes transformers_stream_generator einops" >> ${AUX_INSTALL_SCRIPT}
0 commit comments