Skip to content

Commit 7bf3287

Browse files
authored
Update action.yml (#1499)
1 parent 359518b commit 7bf3287

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

.github/actions/pt2e/action.yml

+3-5
Original file line numberDiff line numberDiff line change
@@ -49,23 +49,21 @@ runs:
4949
fi
5050
# deps
5151
if [[ ${{ inputs.scenario }} == *"performance"* ]]; then
52-
pip install pyyaml botocore
5352
if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then
5453
rm -rf pt2e-audio
5554
git clone --single-branch -b main https://github.com/pytorch/audio pt2e-audio
5655
cd pt2e-audio && git checkout $TORCHAUDIO_COMMIT_ID
57-
python setup.py bdist_wheel && pip uninstall torchaudio -y && pip install --no-deps dist/*.whl
56+
python setup.py bdist_wheel && pip uninstall torchaudio -y && pip install dist/*.whl
5857
cd ../
5958
rm -rf pt2e-vision
6059
git clone --single-branch -b main https://github.com/pytorch/vision pt2e-vision
6160
cd pt2e-vision && git checkout $TORCHVISION_COMMIT_ID
62-
python setup.py bdist_wheel && pip uninstall torchvision -y && pip install --no-deps dist/*.whl
61+
python setup.py bdist_wheel && pip uninstall torchvision -y && pip install dist/*.whl
6362
cd ../
6463
fi
6564
# torchbench
6665
python -c "import torch, torchvision, torchaudio"
67-
cd pt2e-performance && pip install --no-deps -r requirements.txt
68-
pip install -U transformers tokenizers safetensors
66+
cd pt2e-performance && pip install -r requirements.txt
6967
python install.py --continue_on_fail
7068
cd ../
7169
# deps for torchrec_dlrm

0 commit comments

Comments
 (0)