yindo/rocm-automated
1
0
Fork 0
mirror of https://github.com/BillyOutlast/rocm-automated.git synced 2026-02-04 03:51:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

getting ride of upgrade

Browse Source
This commit is contained in:
John Doe
2025-11-23 22:57:52 -05:00
parent 964d37d8ec
commit 4b2959bc31
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfyui-build/docker-entrypoint.sh
View File

@@ -103,9 +103,9 @@ fi
echo "Using PyTorch index URL: ${PYTORCH_INDEX_URL}"
if echo "${PYTORCH_INDEX_URL}" | grep -q "rocm.nightlies.amd.com"; then
pip install --upgrade --pre torch torchvision torchaudio --extra-index-url ${PYTORCH_INDEX_URL}
pip install --pre torch torchvision torchaudio pytorch-triton-rocm --extra-index-url ${PYTORCH_INDEX_URL}
else
pip install --upgrade --pre torch torchvision torchaudio --index-url ${PYTORCH_INDEX_URL}
pip install --pre torch torchvision torchaudio pytorch-triton-rocm --index-url ${PYTORCH_INDEX_URL}
fi
# TO debug flash_attn issues, temporarily disabling its installation
#pip install --upgrade flash_attn --no-build-isolation