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 @@ -409,8 +409,6 @@ jobs:
409409 pip install -U "huggingface_hub[cli]"
410410 huggingface-cli login --token $SECRET_EXECUTORCH_HF_TOKEN
411411 pip install accelerate sentencepiece
412- # TODO(guangyang): Switch to use released transformers library after all required patches are included
413- pip install "git+https://github.com/huggingface/transformers.git@6cc4dfe3f1e8d421c6d6351388e06e9b123cbfe1"
414412 pip list
415413 echo "::endgroup::"
416414
Original file line number Diff line number Diff line change @@ -17,6 +17,8 @@ pip install --no-use-pep517 "git+https://github.com/pytorch/ao.git@${TORCHAO_VER
1717# Install tiktoken for tokenizer
1818pip install lm_eval==0.4.2
1919pip install tiktoken blobfile
20+ # Restore numpy if >= 2.0
21+ pip install " numpy<2.0"
2022
2123# Call the install helper for further setup
2224python examples/models/llama/install_requirement_helper.py
Original file line number Diff line number Diff line change @@ -135,7 +135,7 @@ def python_is_compatible():
135135 "timm==1.0.7" ,
136136 f"torchaudio==2.5.0.{ NIGHTLY_VERSION } " ,
137137 "torchsr==1.0.4" ,
138- "transformers==4.42.4 " ,
138+ "transformers==4.46.0 " ,
139139]
140140
141141# Assemble the list of requirements to actually install.
You can’t perform that action at this time.
0 commit comments