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 @@ -411,8 +411,6 @@ jobs:
411411 pip install -U "huggingface_hub[cli]"
412412 huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
413413 pip install accelerate sentencepiece
414- # TODO(guangyang): Switch to use released transformers library after all required patches are included
415- pip install "git+https://github.com/huggingface/transformers.git@6cc4dfe3f1e8d421c6d6351388e06e9b123cbfe1"
416414 pip list
417415 echo "::endgroup::"
418416
Original file line number Diff line number Diff line change @@ -16,6 +16,8 @@ pip install "$(dirname "$0")/../../../third-party/ao"
1616# Install tiktoken for tokenizer
1717pip install lm_eval==0.4.5
1818pip install tiktoken blobfile
19+ # Restore numpy if >= 2.0
20+ pip install " numpy<2.0"
1921
2022# Call the install helper for further setup
2123python examples/models/llama/install_requirement_helper.py
Original file line number Diff line number Diff line change @@ -137,7 +137,7 @@ def python_is_compatible():
137137 "timm==1.0.7" ,
138138 f"torchaudio==2.5.0.{ NIGHTLY_VERSION } " if USE_PYTORCH_NIGHTLY else "torchaudio" ,
139139 "torchsr==1.0.4" ,
140- "transformers==4.42.4 " ,
140+ "transformers==4.46.1 " ,
141141]
142142
143143# pip packages needed for development.
You can’t perform that action at this time.
0 commit comments