File tree 5 files changed +7
-19
lines changed
backends/apple/coreml/scripts
5 files changed +7
-19
lines changed Original file line number Diff line number Diff line change 1
1
mpmath==1.3.0
2
- numpy==1.21.3; python_version == '3.10'
3
- numpy==1.23.2; python_version == '3.11'
4
- numpy; python_version >= '3.12'
2
+ numpy==2.0.0; python_version >= '3.10'
5
3
PyYAML==6.0.1
6
4
ruamel.yaml==0.17.32
7
5
sympy==1.12
8
6
timm==0.6.13
9
7
tomli==2.0.1
10
8
torchsr==1.0.4
11
- transformers==4.38.0
9
+ transformers==4.47.1
12
10
zstd==1.5.5.1
13
- pandas==2.0.3; python_version == '3.10'
14
- pandas; python_version >= '3.11'
11
+ pandas==2.2.2; python_version >= '3.10'
15
12
pytest==7.2.0
16
13
pytest-cov==4.1.0
17
14
expecttest==0.1.6
Original file line number Diff line number Diff line change @@ -47,11 +47,7 @@ cmake --build "$COREMLTOOLS_DIR_PATH/build" --parallel
47
47
48
48
echo " ${green} ExecuTorch: Installing coremltools."
49
49
pip install " $COREMLTOOLS_DIR_PATH "
50
- # CoreMLTools have started supporting numpy 2.0,
51
- # but ExecuTorch example model test env is still using older transformers,
52
- # so for now we will need to downgrade numpy to 1.x
53
- # TODO: Remove this numpy downgrade once later transformers starts to be used
54
- pip install numpy==1.26.4
50
+
55
51
STATUS=$?
56
52
if [ $STATUS -ne 0 ]; then
57
53
echo " ${red} ExecuTorch: Failed to install coremltools."
Original file line number Diff line number Diff line change @@ -13,8 +13,6 @@ pip install snakeviz sentencepiece
13
13
# Install tiktoken for tokenizer
14
14
pip install lm_eval==0.4.5
15
15
pip install tiktoken blobfile
16
- # Restore numpy if >= 2.0
17
- pip install " numpy<2.0"
18
16
19
17
# Call the install helper for further setup
20
18
python examples/models/llama/install_requirement_helper.py
Original file line number Diff line number Diff line change @@ -157,7 +157,7 @@ def python_is_compatible():
157
157
"timm==1.0.7" ,
158
158
f"torchaudio==2.6.0.{ NIGHTLY_VERSION } " if USE_PYTORCH_NIGHTLY else "torchaudio" ,
159
159
"torchsr==1.0.4" ,
160
- "transformers==4.46 .1" ,
160
+ "transformers==4.47 .1" ,
161
161
]
162
162
163
163
# pip packages needed for development.
Original file line number Diff line number Diff line change @@ -55,12 +55,9 @@ dependencies=[
55
55
" flatbuffers" ,
56
56
" hypothesis" ,
57
57
" mpmath==1.3.0" ,
58
- " numpy==1.21.3; python_version == '3.10'" ,
59
- " numpy==1.23.2; python_version == '3.11'" ,
60
- " numpy; python_version >= '3.12'" ,
58
+ " numpy==2.0.0; python_version >= '3.10'" ,
61
59
" packaging" ,
62
- " pandas==2.0.3; python_version == '3.10'" ,
63
- " pandas; python_version >= '3.11'" ,
60
+ " pandas==2.2.2; python_version >= '3.10'" ,
64
61
" parameterized" ,
65
62
" pytest" ,
66
63
" pytest-xdist" ,
You can’t perform that action at this time.
0 commit comments