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

disabling flash atten

Browse Source
This commit is contained in:
John Doe
2025-11-23 22:34:50 -05:00
parent 932bbbd321
commit eb489d4f11
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -100,15 +100,15 @@ if [ -z "${PYTORCH_INDEX_URL}" ]; then
PYTORCH_INDEX_URL="https://download.pytorch.org/whl/nightly/rocm7.1"
fi
pip uninstall torch torchvision torchaudio pytorch-triton-rocm
echo "Using PyTorch index URL: ${PYTORCH_INDEX_URL}"
if echo "${PYTORCH_INDEX_URL}" | grep -q "rocm.nightlies.amd.com"; then
pip install --pre torch torchvision torchaudio pytorch-triton-rocm --extra-index-url ${PYTORCH_INDEX_URL}
pip install --upgrade --pre torch torchvision torchaudio pytorch-triton-rocm --extra-index-url ${PYTORCH_INDEX_URL}
else
pip install --pre torch torchvision torchaudio pytorch-triton-rocm --index-url ${PYTORCH_INDEX_URL}
pip install --upgrade --pre torch torchvision torchaudio pytorch-triton-rocm --index-url ${PYTORCH_INDEX_URL}
fi
pip install flash-attn
# TO debug flash_attn issues, temporarily disabling its installation
#pip install --upgrade flash_attn --no-build-isolation
echo "Installing ComfyUI requirements..."
pip install -r requirements.txt
@@ -119,7 +119,7 @@ if [ -f "start.sh" ]; then
./start.sh
else
echo "No start.sh found, creating default startup script..."
echo "python main.py --listen 0.0.0.0 --port 8188 --use-split-cross-attention --use-quad-cross-attention" > start.sh
echo "python main.py --listen 0.0.0.0 --port 8188 --use-quad-cross-attention" > start.sh
chmod +x start.sh
./start.sh
fi