Skip to content

Commit

Permalink
downgrade torch 2.6.0 to 20241113
Browse files Browse the repository at this point in the history
  • Loading branch information
tjtanaa committed Dec 6, 2024
1 parent f34cd33 commit 333d6ba
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/amd-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -64,9 +64,9 @@ jobs:
python3 -m pip install pytest pytest-xdist pytest-rerunfailures pytest-flakefinder pytest-cpp
python3 -m pip uninstall -y torch torchvision
python3 -m pip install --pre \
torch==2.6.0.dev20241122+rocm6.2 \
torch==2.6.0.dev20241113+rocm6.2 \
'setuptools-scm>=8' \
torchvision==0.20.0.dev20241122+rocm6.2 \
torchvision==0.20.0.dev20241113+rocm6.2 \
--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.2
python3 -m pip install triton==3.1.0 transformers==4.46.3
python3 -m pip install -e .[dev]
Expand Down

0 comments on commit 333d6ba

Please sign in to comment.