Skip to content

GitLab

"...text-generation-inference.git" did not exist on "f9cf3456250e420af65e1d813ccee6af749658ad"
Commit e1ed4bd6 authored by Casper's avatar Casper
Browse files

Remove +rocm from pip install

parent 952969fd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
.github/workflows/build.yaml
View file @ e1ed4bd6
...@@ -209,9 +209,9 @@ jobs: ...@@ -209,9 +209,9 @@ jobs:
python -m pip install --upgrade build setuptools wheel python -m pip install --upgrade build setuptools wheel
if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then
python -m pip install torch==2.2.0+rocm57 --index-url https://download.pytorch.org/whl/rocm5.7 python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.7
elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then
python -m pip install torch==2.2.0+rocm56 --index-url https://download.pytorch.org/whl/rocm5.6 python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.6
else else
echo Unknown rocm version for python install echo Unknown rocm version for python install
exit 1 exit 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment