File tree 1 file changed +3
-5
lines changed
1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -49,23 +49,21 @@ runs:
49
49
fi
50
50
# deps
51
51
if [[ ${{ inputs.scenario }} == *"performance"* ]]; then
52
- pip install pyyaml botocore
53
52
if [ "${{ inputs.pytorch }}" != "nightly_wheel" ]; then
54
53
rm -rf pt2e-audio
55
54
git clone --single-branch -b main https://github.com/pytorch/audio pt2e-audio
56
55
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
58
57
cd ../
59
58
rm -rf pt2e-vision
60
59
git clone --single-branch -b main https://github.com/pytorch/vision pt2e-vision
61
60
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
63
62
cd ../
64
63
fi
65
64
# torchbench
66
65
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
69
67
python install.py --continue_on_fail
70
68
cd ../
71
69
# deps for torchrec_dlrm
You can’t perform that action at this time.
0 commit comments