@@ -94,7 +94,8 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
9494 COMMIT_TRANSFORMERS=$( python tools/yaml_utils.py -f dependency_version.yml -d transformers -k commit)
9595 VER_PROTOBUF=$( python tools/yaml_utils.py -f dependency_version.yml -d protobuf -k version)
9696 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)
9899 VER_SETUPTOOLS=$( python tools/yaml_utils.py -f dependency_version.yml -d setuptools -k version)
99100 VER_IPEX_MAJOR=$( grep " VERSION_MAJOR" version.txt | cut -d " " -f 2)
100101 VER_IPEX_MINOR=$( grep " VERSION_MINOR" version.txt | cut -d " " -f 2)
@@ -116,7 +117,8 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
116117 mkdir ${WHEELFOLDER}
117118
118119 # 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} "
120122
121123 echo " #!/bin/bash" > ${AUX_INSTALL_SCRIPT}
122124 if [ $(( ${MODE} & 0x04 )) -ne 0 ]; then
@@ -146,7 +148,8 @@ if [ $((${MODE} & 0x02)) -ne 0 ]; then
146148 fi
147149
148150 # 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}
150153 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}
151154 echo " python -m pip install deepspeed==${VER_DS} " >> ${AUX_INSTALL_SCRIPT}
152155 echo " python -m pip install bitsandbytes transformers_stream_generator einops" >> ${AUX_INSTALL_SCRIPT}
0 commit comments